ubuntu20.04安装cudnn

安装 CUDNN 时，需要首先确保 CUDA 已经安装并配置正确。

1. 下载 CUDNN：在 NVIDIA 的 CUDNN 下载页面上下载适用于 Ubuntu 20.04 的版本。

2. 解压下载的文件：使用以下命令解压文件：

tar -xzvf cudnn-<version>-linux-x64-v<cudnn_version>.tgz

3. 安装 CUDNN：使用以下命令安装 CUDNN：

sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

4. 配置环境变量：在 .bashrc 文件中添加以下行：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

5. 更新环境变量：使用以下命令更新环境变量：

source ~/.bashrc

6. 检查安装是否成功：使用以下命令检查 CUDNN 版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果出现 CUDNN_MAJOR 和 CUDNN_MINOR 的版本号，则安装成功。