查看 CUDA 版本:
cat /usr/local/cuda/version.txt
查看 CUDNN 版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 3
#define CUDNN_PATCHLEVEL 1
输出前三行如上,版本就是7.3.1
升级 CUDNN 版本:
- 下载新版本cudnn(官网https://developer.nvidia.com/rdp/cudnn-archive),本文介绍tgz版本(末尾带“for Linux”的那一项)的安装,因为它通用。
- 解压下载好的压缩包,命令:
tar zxvf 【cudnn压缩包名】.tgz -C 【路径】
解压出一个名为cuda的文件夹,文件夹包含include和lib64两个文件夹。
- 删除旧版本cudnn
sudo rm /usr/local/cuda/include/cudnn.h sudo rm /usr/local/cuda/lib64/libcudnn*
-
放入新版本cudnn
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/lib* /usr/local/cuda/lib64/
-
运行动态链接库管理命令:(不确定需不需要sudo)
ldconfig
-
完成更新,可以运行程序看看
【转载自 https://blog.youkuaiyun.com/u011394059/article/details/78455252】
【转载自 https://www.cnblogs.com/yinyl/p/9274615.html】
【转载自 https://blog.youkuaiyun.com/zong596568821xp/article/details/86098833】