安装cuDNN7.1
同样去英伟达官网的cuDNN下载页面:https://developer.nvidia.com/rdp/cudnn-download,
最新版本是cuDNN7.1.4,有三个版本可以选择,分别面向CUDA8.0, CUDA9.0, CUDA9.2:
查看本机的cuda
$ nvcc -V
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 9.2, V9.2.148
所以下载
Download cuDNN v7.1.4 (May 16, 2018), for CUDA 9.2
点开下载
cuDNN v7.1.4 Library for Linux
会得到
cudnn-9.2-linux-x64-v7.1.tgz
下载完cuDNN7.1的压缩包之后解压,然后将相关文件拷贝到cuda的系统路径下即可:
$ tar -zxvf cudnn-9.2-linux-x64-v7.1.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*