The NVIDIA CUDA Deep Neural Network (cuDNN) is a GPU-accelerated library of primitives for use with deep neural networks.
Prerequisites
- Create an NVIDIA developer account from which to download the cuDNN package matching your NVIDIA CUDA version, and appropriate for your Linux distribution. See https://developer.nvidia.com/cudnn.
- Verify you have installed a vSphere Bitfusion client.
- Verify you have installed NVIDIA CUDA.
Procedure
- Install the NVIDIA cuDNN package by running the command sequence for your Linux distribution.
- Ubuntu version 20.04
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
- CentOS 8 and Red Hat Linux 8
sudo rpm -ivh libcudnn8-8.0.5.39-1.cuda11.0.x86_64.rpm
- Ubuntu version 20.04
- (Optional) To verify that NVIDIA cuDNN is installed, run
ldconfig -p | grep cudnn
.
What to do next
- If you are using CentOS or Red Hat Linux, first you must install Python 3. See How to install Python on CentOS and Red Hat Linux.
- If you are using Ubuntu Linux, you can install TensorFlow, PyTorch, and YOLO.