Estoy intentando instalar CUDA en WSL 2 para ejecutar un proyecto que usa TorchAudio y PyTorch. He realizado los pasos exactamente según la documentación.aquí.
Sin embargo, cuando ejecuto el código requerido, aparece el siguiente error:
RuntimeError: No CUDA GPUs are available
Como referencia, tengo una NVIDIA RTX 2060 y mi controlador NVIDIA también está instalado correctamente. ¿Existen próximos pasos recomendados/soluciones conocidas para esto?
Alguna información adicional: estoy usando la última versión preliminar de Windows como se esperaba y mi versión de Ubuntu es 20.04 (descargada directamente de la tienda de Microsoft).
Respuesta1
Suponiendo que tiene instalado el kit de herramientas cuda y todo está en su lugar (nvidia-smi en WSL funcionando y todo). Degradar su versión de pytorch de 1.10 a 1.8.2LTS podría ayudar, esto funcionó para mí, los intentos anteriores que otros informaron que funcionaron para ellos (pero no para mí) fueron degradar su controlador de pantalla NVidia a 472.12 en lugar de 5xx.
pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html