1、查询CUDA版本
输入命令:cat /usr/local/cuda/version.txt
如果CUDA安装路径不是/usr/local/cuda,则需要修改成你的CUDA安装路径
2、查询CUDNN版本
方法一:找到你的CUDNN库文件所在的路径,我的是/usr/local/lib,查看文件libcudnn.so最终链接的文件名,文件名中包含版本号
方法二:
输入命令:cat /usr/local/include/cudnn.h | grep CUDNN_MAJOR -A 2
grep CUDNN_MAJOR -A 2 的意思是匹配CUDNN_MAJOR和后面2行,结果显示CUDNN_MAJOR, CUDNN_MINOR, CUDNN_PATCHLEVEL,与方法一得到的结果应该是一样的。