Спочатку потрібно знайти встановлений файл 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 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
USE_CUDNN
ввімкненого?