Cómo instalar NVIDIA cuDNN en vSphere Bitfusion

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

• Si utiliza CentOS o Red Hat Linux, primero debe instalar Python 3. Consulte Cómo instalar Python en CentOS y Red Hat Linux.
• Si utiliza Ubuntu Linux, puede instalar TensorFlow, PyTorch y YOLO.