☰

在 WSL 2 上安裝 CUDA 後發生錯誤 - RuntimeError: No CUDA GPUs are available

tag-icon • tag-icon ubuntu windows-subsystem-for-linux cuda
在 WSL 2 上安裝 CUDA 後發生錯誤 - RuntimeError: No CUDA GPUs are available

我正在嘗試在 WSL 2 上安裝 CUDA 以運行使用 TorchAudio 和 PyTorch 的專案。我已經完全按照文檔完成了步驟這裡。

但是，當我運行所需的程式碼時，出現以下錯誤：

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 顯示驅動程式降級到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

相關內容