安装CUDA10配套cuDNN,给需要的人参考下。
参考:官网原版安装指南
作者:Su Liang,2018-10-19
cuDNN是nvidia开发的deep neural network library(dnn),即为深度神经网络GPU加速库。
提供了前向/后向/卷积/池化/标准化/激活等层。
(1)注册
https://developer.nvidia.com/rdp/form/cudnn-download-survey
(2)选择cuDNN版本:需要跟CUDA的版本匹配,由于安装的是CUDA10,需选择cuDNN v7.3.1
这里选择了cuDNN v7.3.1 library for linux。没有选择runtime library forubuntu 16.04,安装说明里说,cuDNN v7.3.1 lib for linux这种tar文件(下载的是tgz压缩文件)适用于所有linux系统,而debian文件适用于ubuntu系统。
(3)安装前提:
- GPU
- x86_64 – ubuntu 14.04 or ubuntu 16.04
- CUDA以及nvidia驱动
(4)进入下载目录,解压缩cuDNN安装文件
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
(5)拷贝如下解压缩文件到CUDA toolkit的目录中,然后变更文件允许
不要遗漏命令最后那个*
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*
即完成安装了。
(6)查证是否安装成功
nvcc -V