win10/11下wsl2安装gpu版的pytorch（避坑指南）

0x00 注意

• 不想折腾的不要弄了，老老实实用windows，现在WSL坑还很多。
• 想安装的一定要看官方文档！！在文末
• 本教程只说明在安装了 WSL2 后，并且默认系统是 win11 下安装中的一些坑，当然也可以帮助你快速排查问题。
• 你可能会遇到很多问题，但是这就是你尝试新鲜事物的代价喽，不过欢迎留言区评论~
• 最后成功的环境是win11 、GPU Driver 版本是512.15、CUDA 版本是11.6、 WSL2 Linux内核版本是5.10.102.1
• 本文对 miniconda 和 pytorch 篇幅较少，可以通过查阅其他博客自行安装。

0x01 一定要先检查硬性版本条件

大多数人都会死在这里的第一步，原因是没有检查 wsl2 使用 gpu 所需要的硬性环境到底是什么。然后后面根本装不上。

1. GPU Driver 版本

注意：这里的GPU driver是安装在windows上的，不是wsl中

• 直接升级到最新版本的吧，460多，470多的版本太老旧了。我理解的官方文档是说需要471.21 版本之后的才可以。
• 我这里升级完是 512.15版本，匹配 CUDA 11.6版本的
• 升级步骤
  • 我是有这个NVIDIA GeForce Experience的，所以按照图中的首先选到驱动程序，然后直接检查更新文件即可，安装的时候使用默认的安装方式便可。
    
  • 没有的话就直接到 NVIDIA 官方驱动地址 查询下载。

2. WSL