Ubuntu16.04+cuda8.0+cudnn v6+tensorflow1.4


声明:本文转载自https://my.oschina.net/u/3689493/blog/1584808,转载目的在于传递更多信息,仅供学习交流之用。如有侵权行为,请联系我,我会及时删除。

深度学习第一篇-----tensorflow环境搭建

   学习深度学习首先需要搭建开发环境,现在深度学习框架有很多,但是我还是选择google开源的tensorflow,安装相对容易一点,适合初学者。由于我是做研究用的,所以我下血本配置了一台GTX1060-6g的显卡。由于我的电脑有显卡,所以我选择了装支持gup版本的tensorflow。如果不需要装支持gpu版本的可以参考  http://blog.csdn.net/keith_bb/article/details/74066691这篇文章来安装。由于我买的带有GPU的本子,我想不能浪费我的本子,所以我爬了无数坑装好了支持gup版本的tensorflow。下面我就分享一下我的安装过程。本次是用源码来进行安装的。

环境框架:Ubuntu16.04+cuda8.0+cudnn v6+tensorflow1.4

step1:

下载所需的软件包。

1、下载系统镜像文件Ubuntu16.04  地址https://www.ubuntu.com/download/alternative-downloads

                                                  

如图所示下载镜像文件。

2、下载所需的cuda8.0(gtx1060也支持cuda8.0)

地址https://developer.nvidia.com/cuda-downloads

现在官网直接更新到9.0

cuda 8.0地址https://developer.nvidia.com/cuda-80-ga2-download-archive

3、下载cudnn v6

下载地址:https://developer.nvidia.com/cudnn

下载时可能需要登录,注册账号后填写调差问卷就可以下载。

下载好所需要的安装包就可以进行按住安装了。

step2:安装

1、首先安装显卡GPU驱动

      打开终端:sudo apt-get update

      选择系统设置→软件更新→附加驱动→选择nvidia最新驱动→应用更改.

2、安装Tensorflow依赖的编译工具bazel

      bazel安装方法网址:https://bazel.build/versions/master/docs/install-ubuntu.html

(1) 安装bazel前,需先安装JDK8

      sudo apt-get installsoftware-properties-common

      sudoadd-apt-repository ppa:webupd8team/java

      sudo apt-get update

      sudo apt-get installoracle-java8-installer

      验证java版本:java -version

(2) 安装bazel

      echo "deb [arch=amd64]http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee/etc/apt/sources.list.d/bazel.list

      sudo apt install curl

      curlhttps://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

      sudo apt-get update

      sudo apt-get upgrade bazel

3、由于cuda8.0不支持gcc 5.0以上的编译器,因此需要降级,把编译器版本降到4.9:

       先查看自己的gcc版本

        gcc –v

    降低gcc版本:

      sudoapt-get install g++-4.9

      sudoupdate-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20

      sudoupdate-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10

      sudoupdate-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20

      sudoupdate-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10

      sudoupdate-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

      sudoupdate-alternatives --set cc /usr/bin/gcc

      sudoupdate-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30

      sudoupdate-alternatives --set c++ /usr/bin/g++

查看gcc版本

4、安装cuda 8.0

切换目录到cuda源文件下:

sudo dpkg -icuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

 sudo apt-get install cuda

5、安装cuDNN 6.0

cp cudnn-8.0-linux-x64-v6.0.solitairetheme8 cudnn-8.0-linux-x64-v6.0.tgz

 tar -xvf cudnn-8.0-linux-x64-v6.0.tgz

 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*

6、配置环境变量

 sudo gedit ~/.bashrc

 export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”

  export CUDA_HOME=/usr/local/cuda

  exportPATH="$CUDA_HOME/bin:$PATH"

继续在terminal中输入:

source ~/.bash_profile #使更改的环境变量生效

7、安装Tensflow

(1) 安装Tensorflow依赖的其它工具包

      sudo apt-get install python-numpy swigpython-dev python-wheel

(2) 下载最新的Tensorflow源码

      sudo apt-get install git

      git clone https://github.com/tensorflow/tensorflow

(3) 运行configure脚本配置环境信息(一路按回车建默认就行)或者参照下图配置。

8、 通过bazel来编译pip的安装包,然后通过pip安装

在terminal中输入以下命令:(还是在tensorflow目录下)

1.bazel build -c opt //tensorflow/tools/pip_package:build_pip_package  2.bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package  3.bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg(编译后的安装包在这) 此处可能会出现问题,直安装setuptools-18.1参考链接  4.sudo pip install /home/***(你自己的用户名)/Desktop/tensorflow-1.4.0-cp2-none-any.whl

上述第4部可能会出错,需要安装pip 包

命令:

如图所示 安装成功!!!!

 

 

 

 

 

本文发表于2017年12月05日 00:33
(c)注:本文转载自https://my.oschina.net/u/3689493/blog/1584808,转载目的在于传递更多信息,并不代表本网赞同其观点和对其真实性负责。如有侵权行为,请联系我们,我们会及时删除.

阅读 3262 讨论 0 喜欢 0

抢先体验

扫码体验
趣味小程序
文字表情生成器

闪念胶囊

你要过得好哇,这样我才能恨你啊,你要是过得不好,我都不知道该恨你还是拥抱你啊。

直抵黄龙府,与诸君痛饮尔。

那时陪伴我的人啊,你们如今在何方。

不出意外的话,我们再也不会见了,祝你前程似锦。

这世界真好,吃野东西也要留出这条命来看看

快捷链接
网站地图
提交友链
Copyright © 2016 - 2021 Cion.
All Rights Reserved.
京ICP备2021004668号-1