在 WSL2 上安装 CUDA

最近在写一些小东西的时候需要用到显卡, 但是 Windows 的开发体验又很差 (尤其是对于 Python), 所以就在 WSL 上安装了 CUDA 以供使用.

TIP提示

本文环境:

  • OS: WSL2 Debian bookworm
  • GPU: RTX 3060 Laptop

更新驱动

CUDA 所需要的驱动版本一般挺新的, 所以需要在 Windows 上更新显卡驱动. 使用 Geforce Experience 或者去 官网下载页面 寻找符合你型号的驱动版本即可.

WARNING注意

注意是直接在 Windows 上更新, 不要跑去 WSL 里安装英伟达那恶心的 Linux 驱动了.

更新完成后重启 WSL, 使用 nvidia-smi 查看信息:

0

安装 CUDA Toolkit

英伟达下载页面 选择 WSL-Ubuntu 发行版, 选择 runfile (local) 进行下载.

bash
wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run

TIP提示

留意你所需要的版本, 历史版本可去 归档页 下载, 本文下载的是 12.4 版本.

下载完成后运行安装:

bash
sudo sh cuda_12.4.1_550.54.15_linux.run

安装过程中会询问 EULA, 输入 accept 即可.

WARNING注意

不要更改默认安装选项, 同意 EULA 后应直接开始安装. 否则大概会卡住…

安装完成后, 更新 PATHLD_LIBRARY_PATH, 将下述内容添加到你的配置文件里 (如 ~/.bashrc~/.zshrc):

bash
export PATH="/usr/local/cuda-12.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH"

重启终端使配置生效, 使用 nvcc --version 查看版本信息:

1

安装 cuDNN

英伟达下载页面 下载 cuDNN, 选择与你环境相符的版本下载. 本文下载 cuDNN 9.2.0 for CUDA 12.4:

bash
wget https://developer.download.nvidia.com/compute/cudnn/9.2.0/local_installers/cudnn-local-repo-debian12-9.2.0_1.0-1_amd64.deb
sudo apt install ./cudnn-local-repo-debian12-9.2.0_1.0-1_amd64.deb

添加 keyring 和源:

bash
sudo cp /var/cuda-repo-debian12-9-2-local/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo sed -i "s|^Components: main$|& contrib|" /etc/apt/sources.list.d/cudnn-local-debian12-9.2.0.list

注意根据 apt 的输出修改上述两行命令.

安装 cuDNN:

bash
sudo apt update
sudo apt install -y cudnn-cuda-12

Q.E.D.
部署跨多数据中心分布的 MongoDB 副本集
常轨脱离 Misanthrope