ubuntu下查看cudnn版本+cuda版本

1.查看cudnn版本

首先进入usr目录

cd /usr

查找cudnn.h文件

sudo find -name cudnn.h

运行如下代码

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

若无反应，则查找cudnn_version.h文件（好像因为现在cudnn的版本号已经不在上述文件中了，而是单独有一个文件）

sudo find -name cudnn_version.h

运行如下代码

cat ./include/cudnn_version.h | grep CUDNN_MAJOR -A 2

显示的805则代表下载10.1时候的如下图的8.0.5

2.查看cuda版本

查看驱动的driverAPI最高支持版本

nvidia-smi

查看运行Runtime的API，选择如下任一

nvcc -V

cat /usr/local/cuda/version.txt

可以看出后两者查出来一致。

运行程序时所使用的是运行API，平时装环境要求的cuda，cudnn，tensorflow-gpu版本之间的对应，是针对运行API来讲的，也就是后两者查看的版本。

本文链接：https://www.kjpai.cn/news/2024-04-17/159804.html，文章来源：网络cs，作者：焦糖，版权归作者所有，如需转载请注明来源和作者，否则将追究法律责任！