Сначала вам нужно найти установленный файл cudnn, а затем проанализировать этот файл. Чтобы найти файл, вы можете использовать:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
Если это не сработает, смотрите раздел «Redhat распределения» ниже.
Найдя это место, вы можете сделать следующее (заменив ${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 установите 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"
указывает, что установлена версия 6.0.21 CuDNN.
Redhat распределения
На CentOS я нашел местоположение CUDA с:
$ whereis cuda
cuda: /usr/local/cuda
Затем я использовал процедуру, описанную в файле cudnn.h, который я нашел в этом месте:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
USE_CUDNN
включенным и без него ?