您的位置:首页 > 运维架构 > Linux

在linux(centos)上从源码安装tensorflow

2016-05-13 19:16 453 查看
我参考英文文档在centos7虚拟机上安装了tensorflow:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md#installing-from-sources

具体步骤如下:

安装bazel

Google只提供了在ubuntu下的安装,在centos下的安装可以参考http://www.guudo.cn/tech/view/id/12567 下面主要说一下我遇到的一些问题

configure: error: no acceptable C compiler found in $PATH。这个需要安装gcc:

sudo yum -y install gcc sudo

yum -y install gcc-c+


需要安装JDK1.8,在Oracle官网下载后安装:

sudo rpm -ivh jdk-8u91-linux-x64.rpm


export JAVA_HOME=/usr/java/jdk1.8.0_91

java.lang.OutOfMemoryError: Java heap space 这个需要调整max heap size,在scripts/bootstrap/compile.sh 文件中给javac 命令加上参数:

javac -J-Xmx700m


安装依赖:

sudo yum -y install numpy swig python-devel python-wheel

sudo yum -y install zlib zlib-devel


安装配置:

./configure


编译:

bazel build -c opt --jobs 1 //tensorflow/cc:tutorials_example_trainer


这里我加了参数 –jobs 1。否则bazel默认是200,由于我的物理内存有限,起200个job,导致内存耗尽,系统不断写入写出虚拟内存,编译非常慢。

创建pip包并安装:

bazel build -c --jobs 1 opt //tensorflow/tools/pip_package:build_pip_package


bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg


出现错误:
cp: cannot stat ‘bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/tensorflow’: No such file or directory


直接打开文件 build_pip_package修改路径成
.runfiles/__main__/


sudo pip install /tmp/tensorflow_pkg/tensorflow-0.7.1-cp27-none-linux_x86_64.whl


大功告成:

[cos@localhost git]$ python
Python 2.7.5 (default, Nov 20 2015, 02:00:19)
[GCC 4.8.5 20150623 (Red Hat 4.8.5-4)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
>>>
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow centos 源码