Primero necesita encontrar el archivo cudnn instalado y luego analizar este archivo. Para encontrar el archivo, puede usar:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
Si eso no funciona, vea "Distribuciones de Redhat" a continuación.
Una vez que encuentre esta ubicación, puede hacer lo siguiente (reemplazar ${CUDNN_H_PATH}
con la ruta):
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
El resultado debería verse así:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
Lo que significa que la versión es 7.5.0.
Ubuntu 18.04 (a través de sudo apt install nvidia-cuda-toolkit)
Este método de instalación instala cuda en / usr / include y / usr / lib / cuda / lib64, por lo tanto, el archivo que debe consultar está en /usr/include/cudnn.h.
CUDNN_H_PATH=/usr/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Debian y Ubuntu
A partir de CuDNN v5 en adelante (al menos cuando realiza la instalación a través de sudo dpkg -i <library_name>.deb
paquetes), parece que podría necesitar usar lo siguiente:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
Por ejemplo:
$ 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"
indica que CuDNN versión 6.0.21 está instalada.
Distribuciones Redhat
En CentOS, encontré la ubicación de CUDA con:
$ whereis cuda
cuda: /usr/local/cuda
Luego usé el procedimiento sobre el archivo cudnn.h que encontré desde esta ubicación:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
USE_CUDNN
habilitado?