Ubuntu18.04安装CUDA11.3和cuDNN8.2.0

今天在服务器上跑代码，发现报错，说是CUDA版本不对，然后看了一下服务器的版本，发现是9.0，这就有问题了啊，3090的显卡得用11.0上的版本啊，所以接着配置一下深度学的环境，记录一下方便以后查阅。

CUDA11.3

官网地址：https://developer.nvidia.com/cuda-toolkit-archive

下载自己想要的版本：

根据自己电脑的配置选择相应的选项：

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
1

找到下载文件，赋予运行权限

sudo chmod a+x cuda_11.3.0_465.19.01_linux.run
1

运行安装文件

sudo ./cuda_11.3.0_465.19.01_linux.run
1

输入：accept

因为我们之前已经安装好显卡驱动了，所以这里我就不选择安装：

设置环境变量

sudo vim ~/.bashrc
1

export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
1
2

保存文件，让环境变量生效

source ~/.bashrc
1

验证CUDA Toolkit

nvcc -V
1

cuDNN8.2.0

官网地址：https://developer.nvidia.com/rdp/cudnn-archive

下载下面红框中的四个文件：

依次执行一下命令：

tar -zxvf cudnn-11.3-linux-x64-v8.2.0.53.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

sudo dpkg -i libcudnn8_8.2.0.53-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.0.53-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.0.53-1+cuda11.3_amd64.deb
1
2
3
4
5
6
7
8
9

拷贝例程到用户目录

cp -r /usr/src/cudnn_samples_v8/ $HOME
1

转到minist例程

cd  $HOME/cudnn_samples_v8/mnistCUDNN
1

编译例程

sudo make clean && make
1

编译mnistCUDNN时出错：fatal error: FreeImage.h: No such file or directory

sudo apt-get install libfreeimage3 libfreeimage-dev
1

重新编译后，运行

sudo make clean && make
1

./mnistCUDNN
1

如果cudnn被恰当的安装，将输出打印消息：Test passed!

cudnn 检测版本号

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
1
2

到此CUDA11.3和cuDNN8.2.0安装成功！！！