一、显卡驱动安装
nvidia-smi
得到如下显示:
由此可以确定版本号:12.2以及以下
但是需要注意的是在cudnn和pytorch下寻找对应版本。
二、cuda安装
先通过官方网站安装对应版本:
CUDA Toolkit Archive | NVIDIA Developer,安装执行:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu1804-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu1804-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
然后修改对应的bashrc:
sudo vim .bashrc
在里面最后一行添加对应代码:
export PATH=/usr/local/cuda/bin:$PATH
#保存退出
source .bashrc
之后就可以查看自己显卡的驱动安装:
nvcc -V
三、cudnn开发
进入官网下载包:
cuDNN Archive | NVIDIA Developer
sudo dpkg -i cudnn-local-repo-ubuntu1804-8.9.0.131_1.0-1_amd64_1.deb
cd /var/cudnn-local-repo-ubuntu1804-8.9.0.131
sudo apt-get install '/var/cudnn-local-repo-ubuntu1804-8.9.0.131/libcudnn8_8.9.0.131-1+cuda12.1_amd64.deb'
sudo apt-get install '/var/cudnn-local-repo-ubuntu1804-8.9.0.131/libcudnn8-dev_8.9.0.131-1+cuda12.1_amd64.deb'
sudo apt-get install '/var/cudnn-local-repo-ubuntu1804-8.9.0.131/libcudnn8-samples_8.9.0.131-1+cuda12.1_amd64.deb'
验证:
cd /usr
sudo find -name cudnn_version.h
cat ./include/cudnn_version.h | grep CUDNN_MAJOR -A 2