Ubuntu18.04安装cuDNN

                                         Ubuntu18.04安装cuDNN

一:安装准备

(1)选择cudnn版本

官网cudnn链接:https://developer.nvidia.com/rdp/cudnn-archive

选择cuda版本对应的cudnn,因为本人装的cuda10,所以接下来围绕cudnn7.5.0展开安装(听说这一款比较赞就没选最新的,哈哈哈,看个人叭,你们也可以选最新的,方法差不多的)

(2)开始下载文件

Ubuntu18.04安装cuDNN_第1张图片

这里提醒一下哈,看清楚自己的Unbuntu版本,就比如:如果你是18.04版本你就需要下载三个文件,一个都别漏掉!!!

(3)开始安装

进入相关文件下载好的文件夹中,打开终端输入:

sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb
                                                                                                  

sudo dpkg -i libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64.deb
                                                                                                   

sudo dpkg -i libcudnn7-doc_7.5.0.56-1+cuda10.0_amd64.deb

如下图所示:

Ubuntu18.04安装cuDNN_第2张图片

Ubuntu18.04安装cuDNN_第3张图片

(4)通过安装我们要知道cudnn的路径是什么,就比如我的cudnn路径是:

/usr/include/cudnn

    接下来我们把.h文件复制到/usr/local/cuda/include文件夹下面,并修改权限:    

sudo cp /usr/include/cudnn.h /usr/local/cuda/include
sudo chmod a+x /usr/local/cuda/include/cudnn.h

二:安装检验

       通过上述步骤我们已经基本上装好了cudnn,接下来我们检验一下。

       打开终端,输入:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

Ubuntu18.04安装cuDNN_第4张图片

 如图所示,我们看见了自己cudnn的版本7.5.0,验证完毕!

三.cudnn知识点

         NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。

         CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。

         想要了解更多的可以看简书:https://www.jianshu.com/p/622f47f94784

 

 

        

 

你可能感兴趣的:(Ubuntu18.04安装cuDNN)