Ubuntu安装cuDNN

前言

本文为《CUDA与TensorRT部署学习笔记》系列中环境配置篇。本文主要从cuDNN安装前准备、安装的整体过程进行记录~

1 安装前说明

确保已经安装了相应版本的 CUDA，并清楚下载了对应版本的 cuDNN。

安装cudnn比较简单，就是下载安装的包，然后解压，然后将文件复制到对应的位置即可。

2 下载 cuDNN

登录官网，选择适用于你系统和 CUDA 版本的 cuDNN 版本。

cudnn官方地址：developer.nvidia.com/cudnn

登录后看到如下图：

​​

​​

3 解压、复制安装

（1）先登录查看官方安装指南
官方地址：docs.nvidia.com/deeplearnin…

你会看到如下图所示：

​​

（2）将下载好的文件压缩包放到指定位置，并进行解压

#记得换成自己版本
tar -xvf cudnn-linux-$arch-8.x.x.x_cudaX.Y-archive.tar.xz

（3）复制文件，并给到对应的权限

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

4 验证安装

进到/usr/local/cuda/lib​ 下面看下ll *cudnn*​，得到结果如图：

​​

证明库文件已经到已经复制到指定路径了~！