linux下安装cudnn

原博客地址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(共三个文件),如下图所示:
linux下安装cudnn
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安装成功,则测试结果如下图所示
linux下安装cudnn

上一篇:基于 Ubuntu 系统安装 CUDA 和 cuDNN


下一篇:tensorflow layout optimizer && conv autotune