1、查看 CUDA 版本:

cat /usr/local/cuda/version.txt

或者:

nvcc -V

2、查看 CUDNN 版本:

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

但有可能会不显示,原因是因为cudnn的版本问题,每个版本cudnn.h文件内容不同导致,如下图两个版本的cudnn:

(1)cudnn-10.0-linux-x64-v7.4.2.24
在这里插入图片描述
该版本只有这一个文件。

(2)cudnn-10.2-linux-x64-v8.0.4.30:

在这里插入图片描述
点开cudnn.h文件就可以发现区别,所以查看这个高版本的cudnn需要改为:

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

在这里插入图片描述
注意:我电脑装了两个cuda和两个cudnn,一定要注意cuda和cudnn版本对应。

问题解决!

Logo

更多推荐