ubuntu上查看cudnn版本及更新

更新tensort版本到7.2.1后要求cudnn版本到8.0.4,之前是7.6，需要升级。

1、查看cuda版本

cat /usr/local/cuda/version.txt

2、查看cudnn版本

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

可以看到，目前版本为7.6.5。

3、cudnn更新

下载地址：https://developer.nvidia.com/rdp/cudnn-archive，ubuntu选下面版本就可以了，推荐以压缩包形式安装。

下载完后解压出一个名为cuda的文件夹，文件夹中有include和lib64两个文件夹。解压：

tar -xzvf cudnn-10.2-linux-x64-v8.0.4.30.tgz # 这里我下载的是cuda10.2+cudnn8.0.4

4、删除旧版本

若之前以压缩包的形式安装，则删除以下两个文件夹中cudnn即可：

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

若之前以deb的形式安装，则删除目录较多：

sudo rm -rf /usr/local/cuda/targets/x86_64-linux/include/cudnn.h
sudo rm -rf /usr/lib/x86_64-linux-gnu/libcudnn*
sudo rm -rf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn*
sudo rm -rf /usr/share/doc/libcudnn*
sudo rm -rf /var/lib/dpkg/info/libcudnn*
sudo rm -rf /usr/share/lintian/overrides/libcudnn*

5、安装新版本

cd进入刚才解压的cuda文件夹：

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

6、建立软连接

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.8.0.4
sudo ln -sf libcudnn.so.8.0.4 libcudnn.so.8
sudo ln -sf libcudnn.so.8 libcudnn.so   
sudo ldconfig  

7、测试验证

测试cudnn版本，得到输出如下：

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

输出为空，查看cudnn.h，并未定义版本信息。实测已OK。