He buscado en muchos lugares, pero TODO lo que obtengo es CÓMO instalarlo, no cómo verificar que esté instalado. Puedo verificar que mi controlador NVIDIA esté instalado y que CUDA esté instalado, pero no sé cómo verificar que CuDNN esté instalado. La ayuda será muy apreciada, gracias!
PD.
Esto es para una implementación caffe. Actualmente todo funciona sin CuDNN habilitado.
cuda
computer-vision
caffe
conv-neural-network
cudnn
Alfredox
fuente
fuente
USE_CUDNN
habilitado?Respuestas:
Instalar CuDNN solo implica colocar los archivos en el directorio CUDA. Si ha especificado las rutas y la opción CuDNN correctamente durante la instalación de caffe, se compilará con CuDNN.
Puedes verificar eso usando
cmake
. Crea un directoriocaffe/build
y correcmake ..
desde allí. Si la configuración es correcta, verá estas líneas:Si todo está correcto, simplemente ejecute las
make
órdenes para instalar caffe desde allí.fuente
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
La instalación de CuDNN solo está copiando algunos archivos. Por lo tanto, para verificar si CuDNN está instalado (y qué versión tiene), solo necesita verificar esos archivos.
Instalar CuDNN
Paso 1: Registre una cuenta de desarrollador de nvidia y descargue cudnn aquí (aproximadamente 80 MB). Es posible que necesite
nvcc --version
obtener su versión de cuda.Paso 2: comprueba dónde está tu instalación de cuda. Para la mayoría de las personas, lo será
/usr/local/cuda/
. Puedes consultarlo conwhich nvcc
.Paso 3: copia los archivos:
Comprobar versión
Puede que tenga que ajustar el camino. Vea el paso 2 de la instalación.
Notas
Cuando recibes un error como
con TensorFlow, podría considerar usar CuDNN v4 en lugar de v5.
Usuarios de Ubuntu que lo instalaron a través de
apt
: https://askubuntu.com/a/767270/10425fuente
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
/usr/local/cuda/**/*.h
Primero necesita encontrar el archivo cudnn instalado y luego analizar este archivo. Para encontrar el archivo, puede usar:
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):El resultado debería verse así:
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.
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:Por ejemplo:
indica que CuDNN versión 6.0.21 está instalada.
Distribuciones Redhat
En CentOS, encontré la ubicación de CUDA con:
Luego usé el procedimiento sobre el archivo cudnn.h que encontré desde esta ubicación:
fuente
Para verificar la instalación de CUDA, ejecute el siguiente comando , si está instalado correctamente, el siguiente comando no arrojará ningún error e imprimirá la versión correcta de la biblioteca.
Para verificar la instalación de CuDNN, ejecute el siguiente comando , si CuDNN está instalado correctamente, no obtendrá ningún error.
O
puede ejecutar el siguiente comando desde cualquier directorio
debería dar salida a algo como esto
fuente
nvcc
y verifique su salida. consulte el siguiente enlace también devtalk.nvidia.com/default/topic/457664/…Obteniendo la versión cuDNN [Linux]
Use lo siguiente para encontrar la ruta para cuDNN:
Si lo anterior no funciona, intente esto:
Obteniendo la versión cuDNN [Windows]
Use lo siguiente para encontrar la ruta para cuDNN:
Luego use esto para volcar la versión del archivo de encabezado,
Obteniendo la versión CUDA
Esto funciona tanto en Linux como en Windows:
fuente
cudnn.h
archivo en mi instalación de cuda y pensé que no tenía cudnn instalado. Pero luego ejecuté el código de muestra de cuda descargado del sitio web oficial, y pasó ...type "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h" | findstr "CUDNN_MAJOR CUDNN_MINOR CUDNN_PATCHLEVEL"
Al instalar en ubuntu a través de
.deb
, puede usarsudo apt search cudnn | grep installed
fuente
Correr
./mnistCUDNN
en/usr/src/cudnn_samples_v7/mnistCUDNN
Aquí hay un ejemplo:
fuente
mnistCUDNN
no debería estar en ese directorio ya que no se supone que sea un directorio de escritura. Por el contrario, las muestras deberían haberse copiado como un subdirectorio en el directorio de inicio de los usuarios y haberse creado allí. Entonces, si se instaló y construyó correctamente de acuerdo con las instrucciones en el sitio de Nvidia,mnistCUDNN
estará en~/cudnn_samples_v7
cudnn.h
fue instalado en/usr/include/cudnn.h
, no estoy seguro de por qué, pero esto sucedió antes cuando también instalo CUDAapt-get
. Esta vez estaba usandodpkg
y no