您的位置:首页 > 其它

ubantu16.04下源码编译安装tensorflow

2017-09-14 10:52 381 查看

Ubuntu16.04源码编译安装CPU版本的tensorflow

安装说明:本教程针对tensorflow源码编译安装cpu版本,参考于tensorflow官网

安装环境说明:

os:Ubuntu 16.04

arch:x86_64

tensorflow版本:1.0.1

python版本:python2

tensorflow的一些知识可以访问官网

1.打开terminal,输入:

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

下载tensorflow的源码包

2.输入命令:
$ cd tensorflow
$ git checkout Branch # where Branch is the desired branch
$ git checkout r1.0
说明:这里r1.0代表的是tensorflow的版本,读者可以根据自己需要进行更改。
3.接下来准备好linux下的编译环境
-Bazel
-python tensorflow 依赖
(1)安装bazel
依次输入命令:$ sudo apt-get install openjdk-8-jdk
   		      $ echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
   		      $ curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
$ sudo apt-get update && sudo apt-get install bazel

(2)安装tensorflow依赖

输入命令:$ sudo apt-get install python-numpy python-dev python-pip python-wheel

(如果使用的是Python3:则改为 $ sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel)

(3)安装支持tensorflow GPU的nvidia套件(具体见nvidia官网)

注:电脑gpu的计算能力必须大于或等于3.0。GPU的计算能力表可以取nvidia官网查

(4)输入命令:$ sudo apt-get install libcupti-dev 

4.到这里环境已经配置好了,接下来就可以源码编译安装了。

参考网站:https://www.tensorflow.org/install/install_sources#ConfigureInstallation(需翻墙)

5.$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 

6.$ sudo pip install /tmp/tensorflow_pkg/tensorflow-1.0.1-py2-none-any.whl(注:这里需根据自己使用的版本进行修改)

7.如果上面没出问题的话,到这里基本就大功告成了。打开python,输入import tensorflow,如果不报错,就说明可以正常使用了!




内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: