Ubuntu16.04 安装 cudnn5.1和安装tensorflow框架的完整教程,干货!!!

首先安利一个ubuntu 16.04来截图的快捷方式:

进入keyboard,之后点击Custom Shortcuts。点击‘+’号,在跳出的窗口中,依次输入“截图”和“ gnome-screenshot -a ”,效果图如下:

之后就可以按快捷键去截图了。

进入正题:

在安装好cuda 8.0 以后,接下来博主我就开始安装cudnn 5.1了,首先我们去官网下载cudnn 5.1的压缩包

网址在这里:cudnn5.1的下载地址(必须要注册以后,才可以下载)

下载完成后,我们就可以开始:

首先进入到我们下载那个文件的地址:这里是:Downloads

查看cudnn 5.1版本:


为了检查cudnn5.1有没有安装好,输入 :cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

如下:

至此,我们的ubuntu 16.04 的显卡驱动、cuda 8.0 、cudnn 5.1上面的内容已经配置好了。这里我们开始去安装tensorflow。

步骤:

1、首先我们先检查我们的显卡驱动、cuda8.0、cudnn 5.1是否安装好了,在前面的blog中我们可以了解,这里我再次写出来:

博主的显卡安装的链接:显卡驱动安装教程

   方法(1)nvidia-settings,效果图如下:

   方法(2) lspci | grep -i nvidia,效果如下:

   方法(3)nvidia-smi,效果如下:


上面的任何一种就行了,如果和博主一样,那么你的显卡驱动装好了。

2、cuda8.0 的安装是否完成,参考博主的上篇blog,有具体的步骤:cuda 8.0(安装和检查)链接

3、检查我们的cudnn 5.1有没有安装好:见该篇blog前面部分,有详细的解析。

上面的步骤1,2,3完成后,我们可以开始安装tensorflow(GPU版本)了。

首先:

sudo apt-get install python3-pip
sudo pip3 install tensorflow-gpu

我们会发现会报错,原因是安装的tensorflow 1.3.0版本与cuda8.0 + cudnn5.1环境不兼容,在python中import后会出现各种错误。

解决方案一:要安装1.2.0或1.2.1版本:

sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp35-cp35m-linux_x86_64.whl

这里使用的tsinghua的镜像,效果如下:



现在测试是否能用:

python3
> import tensorflow
>

解决方案2:你就需要使用cuda 9.0以上,才能使用这个版本的tensorflow


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值