WSL2安装微软官网就有，这里不细讲了，主要说坑。

  • 先更新Windows，版本不低于19044 21H2，否则会有问题。

image-20220613215830054

  • WSL要从1升级到2，只有2才支持英伟达驱动。
  • 升级宿主机的驱动，去NV官网找对应版本的最新驱动。

  • 去CUDA Toolkit Archive | NVIDIA Developer下载与你想要安装的Pytorch版本对应的CUDA-Toolkit，安装完毕之后，nvcc -V有可能还是报没安装CUDA-Toolkit，先去/usr/local/cuda/bin看一眼有没有nvcc的可执行文件，如果有的话打开~/.bashrc，把cuda的bin目录加到PATH，也就是把下面这行加到.bashrc中。

    export PATH=$PATH:/usr/local/cuda/bin

如果没有的话，看看CUDA-Toolkit是不是安装好了。安装参考我下面的命令。

# install cuda toolkit
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run

  • 运行nvidia-smi和nvcc -V这里应该都是有输出的，代表安装成功了。

    配置完毕-Screenshot 2022-06-13 220509

  • 装个cuDNN加速计算（可跳过）cuDNN Download | NVIDIA Developer，下载好之后

    tar -xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

  • 最后去Pytorch官网运行安装命令，测试一下cuda和cudnn是否可用，参考如下命令（请勿直接复制，去官网寻找现版本的命令）。

    # install pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

    测试CUDA-cuDNN-Snipaste_2022-06-13_22-48-38

准备完毕~

Last modification：May 30, 2023
© Allow specification reprint
恰饭环节