🐷安装CUDA

1. 在 Windows 端安装 Cuda On WSL 软件

2. 很多教程推荐在 WSL 中通过 Docker 方式（安装 nvida-docker2 和 tensorflow-docker），我们不采用这种方式，在 WSL Ubuntu 终端中执行：

   $ sudo apt-get update
   $ wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
   $ sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
   $ wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-wsl-ubuntu-11-1-local_11.1.0-1_amd64.deb
   $ sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-1-local/7fa2af80.pub
   $ sudo dpkg -i cuda-repo-wsl-ubuntu-11-1-local_11.1.0-1_amd64.deb
   $ sudo apt-get update
   $ sudo apt-get -y install cuda

   在安装过程中如果报错“404 Not Found [IP: 180.101.196.129 443]”，可以先尝试离线安装：

   $ wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run
   $ sudo sh cuda_11.1.0_455.23.05_linux.run

3. 在 .zshrc 中配置：

   export PATH=$PATH:/usr/local/cuda/bin
   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib:/usr/local/cuda/lib64

4. 安装 PyTorch GPU 版本：

   pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

可以使用以下代码测试是否机器能使用 GPU：

import torch
torch.cuda.is_available()

参考文档：

• https://www.jianshu.com/p/98c8af1d2d33
• https://blog.csdn.net/Ashken/article/details/108974058
• https://www.cnblogs.com/dadream/p/13640143.html
• https://blog.csdn.net/LaughingMei/article/details/109962838
• https://blog.csdn.net/xautzxc/article/details/107610353