Vous devez d'abord trouver le fichier cudnn installé, puis analyser ce fichier. Pour trouver le fichier, vous pouvez utiliser:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
Si cela ne fonctionne pas, consultez «Distributions Redhat» ci-dessous.
Une fois que vous avez trouvé cet emplacement, vous pouvez effectuer les opérations suivantes (en le remplaçant ${CUDNN_H_PATH}
par le chemin):
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Le résultat devrait ressembler à ceci:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
Ce qui signifie que la version est 7.5.0.
Ubuntu 18.04 (via sudo apt installe nvidia-cuda-toolkit)
Cette méthode d'installation installe cuda dans / usr / include et / usr / lib / cuda / lib64, par conséquent le fichier que vous devez consulter se trouve dans /usr/include/cudnn.h.
CUDNN_H_PATH=/usr/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Debian et Ubuntu
À partir de CuDNN v5 (au moins lorsque vous installez via des sudo dpkg -i <library_name>.deb
packages), il semble que vous deviez peut-être utiliser les éléments suivants:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
Par exemple:
$ 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"
indique que la version 6.0.21 de CuDNN est installée.
Distributions Redhat
Sur CentOS, j'ai trouvé l'emplacement de CUDA avec:
$ whereis cuda
cuda: /usr/local/cuda
J'ai ensuite utilisé la procédure à propos du fichier cudnn.h que j'ai trouvé à cet emplacement:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
USE_CUDNN
activé?