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

最新推荐文章于 2024-10-17 19:11:48 发布

weixin_34390996

最新推荐文章于 2024-10-17 19:11:48 发布

阅读量131

点赞数

文章标签： java 人工智能 python

原文链接：https://my.oschina.net/u/3689493/blog/1584808

版权

2019独角兽企业重金招聘Python工程师标准>>>

深度学习第一篇-----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

sudo add-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

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10

sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

sudo update-alternatives --set cc /usr/bin/gcc

sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30

sudo update-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

export PATH="$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（编译后的安装包在这）
此处可能会出现如下问题：
Traceback (most recent call last):
  File "setup.py", line 25, in <module>
    from setuptools import find_packages, setup, Command
ImportError: No module named setuptools
直安装setuptools-18.1参考链接https://www.cnblogs.com/huanbia/p/4725421.html
4.sudo pip install /home/***（你自己的用户名）/Desktop/tensorflow-1.4.0-cp2-none-any.whl

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

命令：sudo apt-get install python-pip

之后执行第四部代码效果如下：