WSL2 Ubuntu子系统安装cuda+cudnn+torch

文章目录

• 前言
• 一、安装cudn+cudnn
• 安装pytorch

前言

确保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