ubuntu18.04安装cudnn

在安装 CUDA 之后，您可以安装 cuDNN 来加速深度学习模型的训练。以下是在 Ubuntu 18.04 上安装 cuDNN 的步骤:

1. 从 NVIDIA 的网站上下载 cuDNN 的压缩包。您需要注册并登录才能访问下载链接。

2. 解压压缩包。

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

3. 将 cuDNN 库复制到 CUDA 安装目录。默认安装目录为 /usr/local/cuda。

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo cp  cuda/include/cudnn.h /usr/local/cuda/include/

4. 设置 cuDNN 的运行时库。

sudo nano /etc/ld.so.conf.d/cudnn.conf

在文件中添加：

/usr/local/cuda/lib64

使用下面的命令使其生效

sudo ldconfig

安装完成，重启你的机器。之后检验一下是否安装成功。

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

这条命令应该会输出 CUDNN 的版本号，如果没有问题的话。