centos7源码安装tensorflow1.4

 

本文为服务器源码安装tensorflow1.4.2记录，服务器为Centos7系统。

 

1. tensorflow1.4-GPU所需配置

 

2. bazel 0.5.4安装

 下载bazel-0.5.4-installer-linux-x86_64.sh并运行，之后配置环境变量，下载链接http://pan.baidu.com/s/1pKLKnUz

./bazel-0.5.4-installer-linux-x86_64.sh

#配置环境变量

vim ~/.bashrc

#在.bashrc文件中增加如下字段

export PATH="$PATH:$HOME/bin"
source /home/user/.bazel/bin/bazel-complete.bash

#保存文件并执行
source ~/.bashrc

 

3. 下载tensorflow源码并编译

参考教程https://tensorflow.google.cn/install/source

#使用git克隆源码
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow

#检出要构建的版本分支
git checkout r1.4

#配置构建
./configure

#编译源码
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
 

 

3.1 源码编译遇到的问题

ERROR1:

解决办法：使用yum安装patch

yum install patch

 ERROR2:

解决办法：头文件stropts.h是POSIX XSR的一部分，因为linux不支持STREAMS，所以缺少这个文件。
在/usr/include目录下创建一个空的stropts.h文件即可。

 

4. 生成安装包并安装

编译完成后，使用如下命令构建一个pip软件包（更改路径生成自己的软件包，下面一条命令会在/tem/tensorflow_pkg路径下生成.whl文件）

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

生成的.whl文件取决于Tensorflow版本和平台，使用pip安装

pip install /tmp/tensorflow_pkg/tensorflow-1.4.2-cp27-cp27mu-linux_x86_64.whl --user
#使用--user配置tensorflow为当前用户使用

至此，源码编译安装tensorflow成功，使用tf.__version__查看版本

p.s  pip安装成功后直接在当前路径下import tensorflow会报错 No module named plateform, cd到其他目录即可