在使用 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 安装： conda install 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 build 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 程序，但它需要完整的诊断数据集才能获得构建。但是，我可以配置的设置中没有这样的选项，并且它显示“某些设置已隐藏或由您的组织管理”。