NVIDIA CUDA Deep Neural Network (cuDNN) est une bibliothèque de primitives accélérée par GPU pour une utilisation avec des réseaux de neurones profonds.
Conditions préalables
- Créez un compte de développeur NVIDIA à partir duquel télécharger le module cuDNN correspondant à votre version de NVIDIA CUDA et adapté à votre distribution Linux. Consultez https://developer.nvidia.com/cudnn.
- Vérifiez que vous avez installé un client vSphere Bitfusion.
- Vérifiez que vous avez installé NVIDIA CUDA.
Procédure
- Installez le module NVIDIA cuDNN en exécutant la séquence de commandes pour votre distribution Linux.
- Ubuntu version 20.04
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
- CentOS 8 et Red Hat Linux 8
sudo rpm -ivh libcudnn8-8.0.5.39-1.cuda11.0.x86_64.rpm
- Ubuntu version 20.04
- (Facultatif) Pour vérifier que NVIDIA cuDNN est installé, exécutez la commande
ldconfig -p | grep cudnn
.
Que faire ensuite
- Si vous utilisez CentOS ou Red Hat Linux, vous devez d'abord installer Python 3. Reportez-vous à la section Installation de Python sur CentOS et Red Hat Linux.
- Si vous utilisez Ubuntu Linux, vous pouvez installer TensorFlow, PyTorch et YOLO.