文章目录
前言
确保Windows系统版本高于windows10 21H2或Windows11,然后在Windows中将显卡驱动升级到最新即可,WSL2已支持对显卡的直接调用。
一、安装cudn+cudnn
配置cuda环境,WSL下的Ubuntu子系统的cuda环境与Linux系统下的cuda环境略有不同,所以下载的本地安装包或者联网安装的源不一样,Nvidia官网的WSL下Linux的安装,选项如下图:
sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get -y install cuda
sudo apt install nvidia-cuda-toolkit
安装pytorch
pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html