Ich versuche, CUDA auf WSL 2 zu installieren, um ein Projekt auszuführen, das TorchAudio und PyTorch verwendet. Ich habe die Schritte genau gemäß der Dokumentation ausgeführtHier.
Wenn ich jedoch den erforderlichen Code ausführe, erhalte ich die folgende Fehlermeldung:
RuntimeError: No CUDA GPUs are available
Zur Information: Ich habe eine NVIDIA RTX 2060 und mein NVIDIA-Treiber ist auch korrekt installiert. Gibt es hierfür empfohlene nächste Schritte/bekannte Lösungen?
Einige zusätzliche Informationen: Ich verwende wie erwartet die neueste Windows-Vorschau-Version und meine Ubuntu-Version ist 20.04 (direkt aus dem Microsoft Store heruntergeladen).
Antwort1
Vorausgesetzt, Sie haben das CUDA-Toolkit installiert und alles ist vorhanden (Nvidia-SMI in WSL funktioniert und alles). Ein Downgrade Ihrer PyTorch-Version von 1.10 auf 1.8.2LTS könnte helfen. Bei mir hat das funktioniert. Frühere Versuche, von denen andere berichten, dass sie bei ihnen funktioniert haben (bei mir aber nicht), bestanden darin, Ihren NVidia-Anzeigetreiber auf 472.12 statt 5xx herunterzustufen.
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