WSL 配置CUDA+PyTorch环境

Pytorch兼容的最高CUDA是11.1, 而CUDA官方文档关于WSL给出的只有CUDA11.5的安装步骤，所以在此整理了一下

提示:WSL还是有很多不兼容 想在自己电脑跑各种linux包还是老老实实装双系统

安装CUDA

根据CUDA11.5官方文档改编：

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 dpkg -i 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 apt-get update
sudo apt-get install -y cuda-toolkit-11-1

更新pip

pip install --upgrade pip

安装PyTorch

摘自Pytorch官方文档：

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

杂项

Git Clone指定branch

git clone -b

Ubuntu拷贝指令

cp (需保证source有操作权限)

卸载不正确的CUDA

sudo apt-get --purge remove "*cuda*" "*cublas*" "*cufft*" "*cufile*" "*curand*" \
 "*cusolver*" "*cusparse*" "*gds-tools*" "*npp*" "*nvjpeg*" "nsight*" 
sudo apt-get --purge remove "*nvidia*"
sudo apt-get autoremove