在使用Python 3.7的WSL2 conda环境中的系统错误上未找到NVIDIA驱动程序

Question

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

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

2. 我使用Python 3.7激活了干净的conda环境

3. 然后运行Pytorch安装： 康达安装pytorch torchvision cudatoolkit = 10.2 -c pytorch

然后发生错误，提示未找到NVIDIA驱动程序。我在Pytorch论坛上遇到了一个帖子，有人确实在类似的设置下运行它：Ubuntu 18.04 + Conda + Pytorch https://discuss.pytorch.org/t/found-no-nvidia-driver-on-your-system-but-its-there/35063/4

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

Answer 1

您的Windows版本是什么？ （运行winver.exe） 您需要运行Windows Insider内部版本20145或更高版本才能在WSL2中使用CUDA。

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

Answer 2

@onoma是正确的，因为原始安装步骤缺少一部分，即Windows Insider构建。

发现该文档具有安装具有Cuda支持的WSL2的完整步骤指南。如果有任何人遇到此职位感兴趣：

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

注意：教育版似乎没有选择加入Windows Insider程序的选项。在我的系统上，我可以在Windows设置中访问Windows Insider程序，但是它需要将Diagnostic数据集设置为full才能获取内部版本。但是，我可以在其中进行配置的设置中没有这样的选项，它显示“某些设置由您的组织隐藏或管理”。