在 Python 3.7 的 WSL2 conda 环境中发现您的系统上没有 NVIDIA 驱动程序错误

Found no NVIDIA driver on your system error on WSL2 conda environment with Python 3.7

我有 Nvidia 1080Ti GPU,我想 运行 WSL2 上的 Pytorch,但出现错误“在您的系统上找不到 NVIDIA 驱动程序”,但我确实安装了 NVIDIA 驱动程序。这是我做的步骤。

  1. 我安装了 WSL2,并通过 GeForce Driver 在 WSL 上为 Cuda 安装了 NVIDIA 驱动程序: https://developer.nvidia.com/cuda/wsl/download

  2. 我用 Python 3.7

    激活了一个干净的 conda 环境
  3. 然后我运行安装Pytorch: conda 安装 pytorch torchvision cudatoolkit=10.2 -c pytorch

然后出现错误说找不到NVIDIA驱动程序。我在 Pytorch 论坛上遇到了一个 post,有人确实在类似的设置中得到了 运行:Ubuntu 18.04 + Conda + Pytorch https://discuss.pytorch.org/t/found-no-nvidia-driver-on-your-system-but-its-there/35063/4

我没有多个 GPU,所以我不知道如何在 WSL2 中识别我的驱动程序。感谢您的任何想法!

你的 Windows 版本是多少? (运行 winver.exe) 您需要 运行 Windows Insider build 20145 或更高版本才能在 WSL2 中使用 CUDA。

如果 /dev/dxg 文件存在,您将知道检测到 gpu。

@onoma 是正确的,原始安装步骤缺少一个部分,即 Windows Insider build。

发现此文档包含安装支持 Cuda 的 WSL2 的完整步骤指南。如果有人看到这个 post 会感兴趣:

https://docs.nvidia.com/cuda/wsl-user-guide/index.html

注意:教育版似乎没有 opt-in Windows Insider Program 的选项。在我的系统上,我可以在 Windows 设置中访问 Windows Insider Program,但它需要将诊断数据设置为完整才能获得构建。但是,我可以配置的设置中没有这样的选项,它显示“某些设置已隐藏或由您的组织管理”。