WSL2安装GPU版Pytorch避坑指南

WSL2安装微软官网就有，这里不细讲了，主要说坑。先更新Windows，版本不低于19044 21H2，否则会有问题。WSL要从1升级到2，只有2才支持英伟达驱动。升级宿主机的驱动，去NV官网找对应版本的最新驱动。去CUDA Toolkit Archive | NVIDIA Developer下载与你想要安装的Pytorch版本对应的CUDA-Toolkit，安装完毕之后，nvcc -V有可...

  • Ryushane
  • June 13, 2022
  • No comments