TorchAudio 및 PyTorch를 사용하는 프로젝트를 실행하기 위해 WSL 2에 CUDA를 설치하려고 합니다. 나는 문서에 따라 정확하게 단계를 수행했습니다.여기.
그러나 필수 코드를 실행하면 다음 오류가 발생합니다.
RuntimeError: No CUDA GPUs are available
참고로 NVIDIA RTX 2060이 있고 NVIDIA 드라이버도 올바르게 설치되어 있습니다. 이에 대해 권장되는 다음 단계/알려진 해결 방법이 있습니까?
몇 가지 추가 정보는 예상대로 최신 Windows 미리보기 빌드를 사용하고 있으며 Ubuntu 버전은 20.04입니다(Microsoft 스토어에서 직접 다운로드함).
답변1
cuda 툴킷이 설치되어 있고 모든 것이 제자리에 있다고 가정합니다(WSL의 nvidia-smi 작동 및 모든 것). pytorch 버전을 1.10에서 1.8.2LTS로 다운그레이드하면 도움이 될 수 있습니다. 이는 저에게 효과적이었습니다. 다른 사람들이 해당 버전에서 작동한다고 보고한 이전 시도(저에게는 그렇지 않음)는 NVidia 디스플레이 드라이버를 5xx 대신 472.12로 다운그레이드하는 것입니다.
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