跟着他来 手把手教你在Windows上安装wsl2 Ubuntu
我已经安装好WSL2
了,接下来是安装Anaconda
;
安装好gpu版本的pytorch了,不敢相信自己的眼睛;
哦,还有重要的cuda没装好,最麻烦的是cuda版本和pytorch版本之间的匹配,好要考虑自己显卡的适配程度。(好比画一只完美的眼睛比较容易,要画好整张脸很难,因为不仅要画细节,还需要协调五官比例、间距)
Pytorch-gpu
CUDA 11.8
cuda + cudann
CUDA Toolkit 12.2
Linux, x86_64, WSL-Ubuntu, 2.0, deb(local)
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/12.2.0/local_installers/cuda-repo-wsl-ubuntu-12-2-local_12.2.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-2-local_12.2.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
Download cuDNN v8.9.4 (August 8th, 2023), for CUDA 12.x
下载Local Installer for Linux x86_64 (Tar)
解压缩:
sudo tar -xvf cudnn** #省略部分按tab自动补全
如果报错,就把解压出来的那一个中间文件(名字为blabla多一个后缀,而且只有1kb
)删除再重新解压。
把解压缩得到的文件拷贝到对应文件夹:
sudo cp -r /lib/* /usr/local/cuda-12.2/lib64
sudo cp -r /include/* /usr/local/cuda-12.2/include
或者分别进入到lib/
还有include/
下,再执行:
sudo cp -r * /usr/local/cuda-12.2/lib64
sudo cp -r * /usr/local/cuda-12.2/include
更改读写权限:
sudo chmod a+r /usr/local/cuda-12.2/include/cudnn*
sudo chmod a+r /usr/local/cuda-12.2/lib64/libcudnn*
检查cudann是否安装成功
cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
用VScode+WSL写代码
在VScode中安装WSL插件,再打开WSL中的文件夹即可;