Install NVIDIA cuDNN

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.

Procedure

1. Install the 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

2. To verify that cuDNN is installed, run ldconfig -p | grep cudnn.

What to do next

• If you are using CentOS or Red Hat Linux, you must install Python 3. See Install Python on CentOS and Red Hat Linux.
• If you are using Ubuntu Linux, you can install TensorFlow. See Install TensorFlow.