Win10下Ubuntu终端安装GPU版PyTorch及nvidia-smi报错求助

核心前提先明确：WSL2的Ubuntu子系统不需要单独安装NVIDIA显卡驱动，子系统通过直通链路调用Windows主机侧的驱动，之前在子系统内purge nvidia、cuda相关包的操作本身不会损坏底层驱动，nvidia-smi通信报错基本都是链路配置问题，按以下顺序排查修复：

• 修复Windows侧基础驱动
  卸载Windows上现有NVIDIA驱动，下载对应RTX 3090的官方正式版驱动（Game Ready/Studio版均可），安装时勾选「清洁安装」选项，安装完成后重启整个Windows主机，不要只重启WSL。重启后先在Windows PowerShell中执行nvidia-smi，确认能正常输出显卡型号、驱动版本号，这一步不通后面所有操作都无效。
• 修复WSL2 GPU直通链路
  打开PowerShell执行wsl --update将WSL内核升级到最新版本，更新完成后执行wsl --shutdown彻底关闭所有运行中的WSL实例，等待10秒后再打开Ubuntu终端。

  注意：必须使用WSL2版本，WSL1不支持GPU直通。在PowerShell执行wsl -l -v可查看已安装发行版的WSL版本，如果显示版本号为1，执行wsl --set-version <你的Ubuntu发行版名称> 2完成版本转换。

• 干净安装GPU版PyTorch
  进入Ubuntu终端后，不要通过apt安装任何nvidia驱动、桌面版CUDA Toolkit包，这类包会覆盖WSL自带的直通适配库，直接安装PyTorch内置的CUDA运行时即可。为避免包版本冲突，先新建独立虚拟环境：

  python3.9 -m venv pt_gpu_env
  source pt_gpu_env/bin/activate
  pip install --upgrade pip setuptools wheel
  

  安装适配CUDA 11.8的PyTorch版本（RTX 30系显卡对CUDA 11.x系列兼容性最优，不建议用过低版本CUDA）：

  pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
  

• 功能验证
  安装完成后先在Ubuntu终端执行nvidia-smi，正常情况下会输出和Windows侧一致的显卡、驱动信息。再进入Python交互环境验证PyTorch GPU可用性：

  import torch
  print(torch.cuda.is_available())
  # 输出True即代表GPU调用正常
  print(torch.cuda.get_device_name(0))
  # 正常输出应为NVIDIA GeForce RTX 3090
  

常见踩坑说明 #

• 如果之前在子系统内通过apt装过nvidia-driver、cuda等包，重新执行purge命令清理后，执行wsl --shutdown重启WSL即可恢复，不需要重装子系统。
• 不要混用第三方源、conda默认源的PyTorch包，这类包经常出现CUDA版本编译不匹配的问题，优先使用官方pip源的预编译whl包安装。
• 如果以上操作后仍报驱动通信错误，检查Windows系统功能是否开启：进入「控制面板-程序-启用或关闭Windows功能」，勾选「适用于Linux的Windows子系统」「虚拟机平台」两个选项，保存后重启系统再重试。

内容的提问来源于stack exchange，提问作者Broke Bum