1. 检查当前系统 CUDA 和 cuDNN 版本信息
1.1 查看系统 CUDA Version信息
cat /usr/local/cuda/version.txt
CUDA Version 信息显示如下,查看 CUDA 的信息是帮助我们后面选择正确的版本的 cuDNN 安装包。
CUDA Version 10.0.130
1.2 查看系统 cuDNN Version信息
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cuDNN Version信息显示如下:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 3
#define CUDNN_PATCHLEVEL 1
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
2. 删除系统 cuDNN
当系统中的 cuDNN 版本不是我们想要的版本的时候,可以先删除当前版本,然后安装我们需要的版本。
使用如下指令删除旧版本的 cuDNN
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn
3. 安装新版本 cuDNN
3.1 下载新版本 cuDNN
cuDNN 安装包下载地址:https://developer.nvidia.com/rdp/cudnn-archive
解压下载的 cuDNN 安装包,并 cuda 文件夹中打开终端,执行如下指令:
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
当然如果系统安装了比如 anaconda3 这样的工具,还要给 anaconda3 路径下 Copy 对应文件,可以使用如下指令,或者直接手动 Copy 到对应路径。
sudo cp -r cudnn /home/Gavin/anaconda3/pkgs/
sudo cp lib64/lib* /home/Gavin/anaconda3/lib/
3.2 建立链接
打开终端,执行如下指令:
cd /usr/local/cuda/lib64
sudo chmod +r libcudnn.so.7.4.2
sudo ln -sf libcudnn.so.7.4.2 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
3.3 检查新版本信息
按照之前的方式检查新的 cuDNN 版本信息,确认是否安装 OK。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cuDNN Version信息显示如下:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 4
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
按照如上步骤,便可将系统 cuDNN 从原来7.3.1的版本更新为7.4.2版本。
网友评论