ubuntu18.04下cuda与cudnn的设置

一、显卡驱动安装

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/keyrin