NVIDIA CUDA Deep Neural Network (cuDNN) es una biblioteca de primitivas con aceleración de GPU para usar con redes neuronales profundas.

Requisitos previos

  • Cree una cuenta de desarrollador de NVIDIA desde la que descargar el paquete de cuDNN que coincida con su versión de NVIDIA CUDA y que sea adecuado para su distribución de Linux. Consulte https://developer.nvidia.com/cudnn.
  • Compruebe que ha instalado un cliente de vSphere Bitfusion.
  • Compruebe que haya instalado NVIDIA CUDA.

Procedimiento

  1. Instale el paquete de NVIDIA cuDNN ejecutando la secuencia de comandos correspondiente a su distribución de Linux.
    • Ubuntu versión 20.04
      sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
    • CentOS 8 y Red Hat Linux 8
      sudo rpm -ivh libcudnn8-8.0.5.39-1.cuda11.0.x86_64.rpm
  2. (opcional) Para comprobar que NVIDIA cuDNN esté instalado, ejecute ldconfig -p | grep cudnn.

Qué hacer a continuación