注意:最好按着文档要求的版本装库。另外,在nvidia官方网站下载库的时候,不稳定……多试……
查看显卡支持的版本
nvidia-smi
安装CUDA
参考:<https://blog.51cto.com/shoucuohulu/6283523>
依赖
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
- 进入 <https://developer.nvidia.com/cuda-downloads> 看到,默认的是12.1版本,不是我们想要的。找到 Archive of Previous CUDA Releases 也就是 <https://developer.nvidia.com/cuda-toolkit-archive> 打开后,找到12.0版本( <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
- 配置环境变量 参考: <https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions>
- 修改
~/.bashrc
export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- 应用
source ~/.bashrc
- 查看是否安装成功
nvcc --version
报错 /sbin/ldconfig.real: /usr/lib/wsl/lib/libcuda.so.1 is not a symbolic link
参考:<https://blog.csdn.net/qq_42756195/article/details/125769622> 原因 /usr/lib/wsl/lib/目录下都是文件而不是链接,且该目录只读,需要在其他目录操作
解决
cd /usr/lib/wsl
sudo mkdir lib2
sudo ln -s lib/* lib2
更改wsl配置文件
sudo vim /etc/ld.so.conf.d/ld.wsl.conf
将 /usr/lib/wsl/lib 改为 /usr/lib/wsl/lib2 测试修改是否生效
sudo ldconfig
永久修改(最好手工合并,不建议执行下面脚本)
sudo cat >> /etc/wsl.conf << EOF
[automount]
ldconfig = fasle
EOF
卸载
<https://blog.csdn.net/davidhopper/article/details/81206673>
cuDNN
下载: <https://developer.nvidia.com/rdp/cudnn-download>
文档: <https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html>
安装
sudo apt-get install zlib1g
# 需先下载
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.1.23_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.1.23/cudnn-local-E7A7D88D-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install libcudnn8
sudo apt-get install libcudnn8-dev
sudo apt-get install libcudnn8-samples
卸载
sudo apt-get remove libcudnn8
rm -rf /usr/share/keyrings/cudnn-local-*
sudo dpkg -P cudnn-local-repo-ubuntu2204-8.9.1.23
TensorRT
文档: <https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html>
# 需先下载
sudo dpkg -i nv-tensorrt-local-repo-ubuntu2204-8.6.1-cuda-12.0_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-ubuntu2204-8.6.1-cuda-12.0/nv-tensorrt-local-42B2FC56-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt