Я пытаюсь установить CUDA на WSL 2 для запуска проекта, который использует TorchAudio и PyTorch. Я выполнил шаги точно по документацииздесь.
Однако при запуске требуемого кода возникает следующая ошибка:
RuntimeError: No CUDA GPUs are available
Для справки, у меня NVIDIA RTX 2060, и мой драйвер NVIDIA установлен правильно. Есть ли какие-либо рекомендуемые дальнейшие шаги/известные решения для этого?
Дополнительная информация: я использую последнюю предварительную сборку Windows, как и ожидалось, а моя версия Ubuntu — 20.04 (скачана напрямую из магазина Microsoft).
решение1
Предполагая, что у вас установлен cuda toolkit и все в порядке (nvidia-smi в WSL работает и все такое). Понижение версии pytorch с 1.10 до 1.8.2LTS может помочь, это сработало для меня, предыдущие попытки, которые, как сообщают другие, сработали для них (но не для меня), заключались в понижении версии драйвера дисплея NVidia до 472.12 вместо 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