cuda和cudnn安装详解

从事深度学习无论是tensorflow还是caffe都需要安装cuda和cudnn这2个显卡支持的库,经过一番倒腾,将经验分享给大家。

cuda的安装

1、下载
cuda首先去官网下载。在选择版本的时候很重要,默认下载是最新的,若想下载旧的版本可点击如下图红色区域:

在这里插入图片描述
在选择版本还是比较重要的,因为显卡驱动和cuda的版本需要匹配,否则显示安装了,后面测试通不过,可以参考:cuda relaease Notes主要是显卡驱动的匹配,或者你在安装cuda的时候安装显卡驱动,但cuda安装的不一定适合电脑的显卡,因此在安装cuda的适合尽量不要让cuda安装驱动,显卡驱动和cuda版本:
在这里插入图片描述
2、安装
下载选择了版本后就会,下面就有安装的命令,如下图。
在这里插入图片描述
Note:下载安装nvidia driver的时候选择no,其他的都选yes或者直接Enter键。
检查是否配置环境:

gedit .bashrc
// 查看是否含有:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH 

export PATH=/usr/local/cuda-8.0/bin: P A T H e x p o r t L D L I B R A R Y P A T H = / u s r / l o c a l / c u d a / l i b 64 : PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64: PATHexportLDLIBRARYPATH=/usr/local/cuda/lib64:LD_LIBRARY_PATH

3、测试
安装好了后就需要测试,cuda的安装是否成功:

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make clean
sudo make
./deviceQuery

如果显示:


Result = PASS

则安装成功。
若显示:

CUDA driver version is insufficient for CUDA runtime version

则表示显卡驱动和cuda版本不匹配。

cudnn安装

cudnn下载
官方下载地址,这个网站可能需要翻墙下载,并且要注册登陆才可以下载。这里是我下载的,由于上传有大小限制分为4部分上传:第一部分第二部分第三部分第四部分。(前2部分是cudnn-9.0-linux-x64-v7.tgz ,将2部分放在一个文件,任一解压一个,即可。后面2个是3个deb文件)
上面我cuda选择的是cuda9.0,ubantu16.04系统,因此下载如下:
在这里插入图片描述
下载后四个文件如下:
在这里插入图片描述
2、安装
解压cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8的后缀名修改为tgz,然后用如下命令解压

$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz

拷贝.h 和 libs文件到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*

安装剩下的三个.deb文件:

#Install the runtime library, for example:
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb

#Install the developer library, for example:
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb

# Install the code samples and the cuDNN Library User Guide, for example:
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

3 测试

# Copy the cuDNN sample to a writable path.
cp -r /usr/src/cudnn_samples_v7/ $HOME

# Go to the writable path.
 cd  $HOME/cudnn_samples_v7/mnistCUDNN

#Compile the mnistCUDNN sample.
make clean && make

#Run the mnistCUDNN sample.
 ./mnistCUDNN

If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:
Test passed!

如果还有不明白的地方,请参考官网

这里位置就安装好了cuda和cudnn。当然这里顺便提一下,tensorflow在安装的时候 也与cuda需要匹配,这中间的坑比较多,目前cuda8.0支持tensorflow1.4版本,cuda9.0支持的是tensorflow1.5版本。

  • 3
    点赞
  • 40
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值