Ubuntu 18.04 切换cudnn版本

1. 查看当前版本

# 8.0以下版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 8.0以上版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

2. 下载希望更新的版本

CUDNN
需要NVIDIA账号登录，例如这里下载的cudnn 8.0.3.33 for cuda 10.2，文件名为cudnn-10.2-linux-x64-v8.0.3.33.tgz.

3. 删除或备份原来的cudnn文件

 sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

4. 解压下载的文件并安装

tar zxvf cudnn-10.2-linux-x64-v8.0.3.33.tgz
# sudo cp cuda/include/cudnn.h /usr/local/cuda/include # old version
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include # new version
sudo cp cuda/lib64/lib* /usr/local/cuda/lib64/

5. 更新软链接

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.8.0.3
sudo ln -sf libcudnn.so.8.0.3 libcudnn.so.8
sudo ln -sf libcudnn.so.8 libcudnn.so
sudo ln -sf libcudnn_adv_train.so.8.0.3 libcudnn_adv_train.so.8
sudo ln -sf libcudnn_adv_infer.so.8.0.3 libcudnn_adv_infer.so.8
sudo ln -sf libcudnn_ops_train.so.8.0.3 libcudnn_ops_train.so.8
sudo ln -sf libcudnn_ops_infer.so.8.0.3 libcudnn_ops_infer.so.8
sudo ln -sf libcudnn_cnn_infer.so.8.0.3 libcudnn_cnn_infer.so.8
sudo ln -sf libcudnn_cnn_train.so.8.0.3 libcudnn_cnn_train.so.8
sudo ldconfig