一、显卡驱动安装

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