本文为服务器源码安装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到其他目录即可