Ubuntu 安装 cuDNN(附测试)

为深度学习所用,博主预想在Ubuntu16.04上安装 显卡驱动 + CUDA + cuDNN + Tensorflow-gpu + Keras + PyCharm,参考了众多资料,最终成功将所有软件安装完毕,且能成功运行使用。该篇博客介绍了cuDNN的安装教程,亦可通过下方链接查看其他步骤的安装教程。

前期说明及版本对照

换源及安装显卡驱动

安装 CUDA(附测试)

安装 cuDNN(附测试)

安装 Tensorflow-gpu 与 Keras(附测试)

设置 PyCharm

安装教程完整版

安装 cuDNN

可在cuDNN的官网上下载对应版本的cuDNN安装包,官网网址:cuDNN Archive

1、安装cuDNN

cuDNN的安装方法有两种,需要下载不同的安装包:

(1).tgz 文件安装

该方法仅需下载一个安装包,如图所示
在这里插入图片描述
下载完毕后,进入文件所在位置,解压安装包(其中 xxx 为安装包的补充名称)

tar -xzvf cudnn-xxx.tgz

而后将cuDNN拷贝,放入CUDA对应的文件夹中,并赋予其权限

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*

(2).deb 文件安装

该方法需要下载三个安装包,如下图所示
在这里插入图片描述
分别对这三个安装包执行以下命令进行安装,文件名请自行修改,要注意,必须按以下顺序进行安装,否则会因依赖问题而报错

sudo dpkg -i libcudnn7_7.6.4.38-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.4.38-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.4.38-1+cuda9.0_amd64.deb

2、测试cuDNN

选择 .deb 文件安装cuDNN,默认在 “Home” 位置,会安装有cuDNN的测试案例

在这里插入图片描述
进入 samples 文件中的 mnistCUDNN 文件夹下,鼠标右键打开终端

在这里插入图片描述
编译文件

make clean && make

编译后执行

./mnistCUDNN

若最后结果显示 Test passed,则说明测试成功

在这里插入图片描述

  • 5
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,cuDNN是NVIDIA专门为深度学习框架提供的加速库,它可以优化卷积神经网络的训练和推理过程。如果你想在Ubuntu上使用cuDNN,需要安装CUDAcuDNN两个软件包。 下面是具体的安装步骤: 1. 安装CUDA 首先,你需要先安装CUDA,可以从NVIDIA的官网上下载对应版本的CUDA安装包。下载完成后,打开终端,进入到安装包所在的目录,执行以下命令进行安装: ``` sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb sudo apt-get update sudo apt-get install cuda ``` 2. 安装cuDNN 下载对应版本的cuDNN安装包,需要注意的是,cuDNN需要与已安装CUDA版本对应。下载完成后,解压缩文件,执行以下命令进行安装: ``` 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* ``` 安装完成后,可以通过以下命令检查cuDNN是否正确安装: ``` nvcc -V && cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 ``` 如果输出结果中显示了CUDAcuDNN的版本信息,则说明安装成功。 3. 配置环境变量 在终端中打开.bashrc文件,在文件末尾添加以下两行: ``` export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda/bin:$PATH ``` 至此,cuDNN安装和配置就完成了。如果你想在深度学习框架中使用cuDNN加速,只需要将对应的库链接到项目中即可。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值