原博客地址https://blog.csdn.net/davidhopper/article/details/81206673
安装并配置CUDNN 7.1 for CUDA 9.0
下载CUDNN 7.1 for CUDA 9.0
下载CUDNN需要在官网注册一个账号,登录后下载CUDNN 7.1 for CUDA 9.0(共三个文件),如下图所示:
1.4.2 安装CUDNN 7.1 for CUDA 9.0
进入CUDNN 7.1 for CUDA 9.0三个文件所在的目录,执行如下命令安装:
cd ~/tools/cudnn/7.1_for_cuda9.0
sudo dpkg -i libcudnn7_7.1.4.18-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.1.4.18-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.1.4.18-1+cuda9.0_amd64.deb
1.4.3 验证CUDNN 7.1 for CUDA 9.0是否安装成功
使用如下命令将示例代码复制到当前用户目录,编译并运行其中的一个示例程序mnistCUDNN:
cp -r /usr/src/cudnn_samples_v7/ ~
cd ~/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
如果CUDNN安装成功,则测试结果如下图所示