最近在写一些小东西的时候需要用到显卡, 但是 Windows 的开发体验又很差 (尤其是对于 Python), 所以就在 WSL 上安装了 CUDA 以供使用.
TIP提示
本文环境:
- OS: WSL2 Debian bookworm
- GPU: RTX 3060 Laptop
更新驱动
CUDA 所需要的驱动版本一般挺新的, 所以需要在 Windows 上更新显卡驱动. 使用 Geforce Experience 或者去 官网下载页面 寻找符合你型号的驱动版本即可.
WARNING注意
注意是直接在 Windows 上更新, 不要跑去 WSL 里安装英伟达那恶心的 Linux 驱动了.
更新完成后重启 WSL, 使用 nvidia-smi 查看信息:
安装 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.runTIP提示
留意你所需要的版本, 历史版本可去 归档页 下载, 本文下载的是 12.4 版本.
下载完成后运行安装:
bash
sudo sh cuda_12.4.1_550.54.15_linux.run安装过程中会询问 EULA, 输入 accept 即可.
WARNING注意
不要更改默认安装选项, 同意 EULA 后应直接开始安装. 否则大概会卡住…
安装完成后, 更新 PATH 和 LD_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 查看版本信息:
安装 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-12Q.E.D.