我是靠谱客的博主 年轻月亮,最近开发中收集的这篇文章主要介绍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


最后

以上就是年轻月亮为你收集整理的Ubuntu16.04 安装 cudnn5.1和安装tensorflow框架的完整教程,干货!!!的全部内容,希望文章能够帮你解决Ubuntu16.04 安装 cudnn5.1和安装tensorflow框架的完整教程,干货!!!所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(41)

评论列表共有 0 条评论

立即
投稿
返回
顶部