美文网首页
Ubuntu: cuDNN 版本更新方法

Ubuntu: cuDNN 版本更新方法

作者: Gavin先生 | 来源:发表于2019-08-20 20:17 被阅读0次
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版本。

相关文章

网友评论

      本文标题:Ubuntu: cuDNN 版本更新方法

      本文链接:https://www.haomeiwen.com/subject/awcmsctx.html