您首先需要找到已安装的cudnn文件,然后解析该文件。要查找文件,可以使用:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
如果那不起作用,请参阅下面的“ Redhat发行版”。
找到此位置后,您可以执行以下操作(将${CUDNN_H_PATH}
替换${CUDNN_H_PATH}
路径):
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
结果应如下所示:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
这意味着版本是7.5.0。
Ubuntu 18.04(通过sudo apt install nvidia-cuda-toolkit)
这种安装方法会将cuda安装在/ usr / include和/ usr / lib / cuda / lib64中,因此您需要查看的文件位于/usr/include/cudnn.h中。
CUDNN_H_PATH=/usr/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Debian和Ubuntu
从CuDNN v5开始(至少在通过sudo dpkg -i <library_name>.deb
软件包安装时),您可能需要使用以下命令:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
例如:
$ cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 6
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
表示已安装CuDNN 6.0.21版。
Redhat发行版
在CentOS上,我找到了CUDA的位置:
$ whereis cuda
cuda: /usr/local/cuda
然后,我在从该位置找到的cudnn.h文件中使用了有关过程:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
0
我已经搜索了很多地方,但是得到的只是如何安装它,而不是如何验证它是否已安装。我可以确认已安装NVIDIA驱动程序,并且已安装CUDA,但我不知道如何验证已安装CuDNN。帮助将不胜感激,谢谢!
PS。
这是针对caffe的实现。当前,所有功能在未启用CuDNN的情况下都可以正常工作。