吃水不忘挖井人,参考:
- Windows10/11 WSL2 安装nvidia-cuda驱动 - 哔哩哔哩 (bilibili.com)
- cudnn (102条消息) Windows 11/10 WSL2 Ubuntu 20.04 下配置Cuda及Pytorch_windows cuda_home_Quatttschen的博客-CSDN博客
查看显卡支持的版本
nvidia-smi
从下面的图片中可以看出,支持的CUDA版本是11.6
安装CUDA
安装文档
NVIDIA CUDA Installation Guide for Linux
依赖liburcu6
wget http://archive.ubuntu.com/ubuntu/pool/main/libu/liburcu/liburcu6_0.11.1-2_amd64.deb
sudo dpkg -i liburcu6_0.11.1-2_amd64.deb
选择WSL
进入 https://developer.nvidia.com/cuda-downloads 看到,默认的是12.1版本,不是我们想要的。找到 Archive of Previous CUDA Releases 也就是https://developer.nvidia.com/cuda-toolkit-archive 打开后,找到11.6.2版本(https://developer.nvidia.com/cuda-11-6-2-download-archive)。
下边会自动生成安装脚本。
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.6.2/local_installers/cuda-repo-wsl-ubuntu-11-6-local_11.6.2-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-6-local_11.6.2-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
说明:开发环境中local方式和network方式都可以,如果是服务器没有互联网的情况下,选择local方式。
配置环境变量
参考: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions
sudo vim ~/.bashrc
export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
查看是否安装成功
nvcc --version
安装pytorch
打开链接 https://pytorch.org/get-started/locally/
选择相关版本
最低支持cuda11.7……
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
最终:
(modelscope) bigdata@Work-for-NGY:~$ pip list|grep torch
torch 1.13.1+cu116
torchaudio 0.13.1+cu116
torchvision 0.14.1+cu116
如果 1.13版本太低,说明当前硬件配置太低了…… 得升级显卡……