吃水不忘挖井人,参考:


查看显卡支持的版本

nvidia-smi

从下面的图片中可以看出,支持的CUDA版本是11.6

Windows WSL2中安装CUDA_CUDA

安装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)。

Windows WSL2中安装CUDA_hive_02

下边会自动生成安装脚本。

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/

选择相关版本

Windows WSL2中安装CUDA_hive_03

最低支持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版本太低,说明当前硬件配置太低了…… 得升级显卡……