Ubuntu 18.04 安装 Tensorflow 环境之三:安装 CUDNN 7.4.1

 

CUDA 10 就安装好了,下来需要安装 cudnn 7.4.1,才能利用 GPU 进行计算加速。

安装过程

1、下载CUDNN

NVIDIA官网上下载 NVIDIA CUDNN 相关资源,NVIDIA 官网需要注册,在 login 进去之后,点击同意“I Agree To the Terms of the cuDNN Software License Agreement”,相关资源才会显示。

在官网上下载4个文件,如果不想进行注册,或需要翻墙才能注册,或由于网速差等不能下载的,我已上传相关资源,可点击每个资源后面的链接,进行下载。

要下载四个文件
cuDNN Library for Linux  国内下载地址
cuDNN Runtime Library for Ubuntu18.04 (Deb)  国内下载地址
cuDNN Developer Library for Ubuntu18.04 (Deb)  国内下载地址
cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)  国内下载地址

2、将下载好的文件,传到Ubuntu服务器上的文件夹,如 tools

3、修改文件名

需要注意的是,下载的最新的Linux版的CUDNN的文件名是 cudnn-10.0-linux-x64-v7.4.1.5.solitairetheme8,我们需要将其修改为 .tgz 结尾的文件:

cp cudnn-10.0-linux-x64-v7.4.1.5.solitairetheme8 cudnn-10.0-linux-x64-v7.4.1.5.tgz

4、打开终端,按顺序安装,注意,一定要按顺序进行安装。

tar -xzvf cudnn-10.0-linux-x64-v7.4.1.5.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

然后安装其他三个包:

sudo dpkg -i libcudnn7_7.4.1.5-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.4.1.5-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-doc_7.4.1.5-1+cuda10.0_amd64.deb

5、验证 CUDNN 是否安装成功

cp -r /usr/src/cudnn_samples_v7/ .
cd cudnn_samples_v7/mnistCUDNN/
 make clean && make
./mnistCUDNN

如输出下图,则说明安装成功。

如果编译的时候出现报错:

./mnistCUDNN: error while loading shared libraries: libcudart.so.10.0: cannot open shared object file: No such file or directory

解决办法:首先确认/etc/profile中的路径包含了cuda10.0的安装路径及相应的库文件

export PATH=$PATH:/usr/local/cuda-10.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.0/lib64

若无,将上面三行加入profile文件末尾:

sudo vi /etc/profile

使配置文件生效,再次执行。 

source /etc/profile

至此,done.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章