☰

在 WSL 2 上安装 CUDA 后出错 - RuntimeError：没有可用的 CUDA GPU

tag-icon • tag-icon ubuntu windows-subsystem-for-linux cuda
在 WSL 2 上安装 CUDA 后出错 - RuntimeError：没有可用的 CUDA GPU

我正在尝试在 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

相关内容