红联Linux门户
Linux帮助

ubuntu15.10源码安装tensorflow

发布时间:2015-11-27 21:46:15来源:linux网站作者:autumnqin

刚买的新机器,安装上最新的ubuntu系统。按照官网的流程安装的,没有选择GPU支持。


下载源码。需要git,如果没有安装git需要先安装。

git clone --recurse-submodules https://github.com/tensorflow/tensorflow


安装Bazel
先安装依赖

sudo apt-get install openjdk-8-jdk openjdk-8-source
sudo apt-get install pkg-config zip g++ zlib1g-dev unzip


下载安装文件:
https://github.com/bazelbuild/bazel/releases/download/0.1.1/bazel-0.1.1-installer-linux-x86_64.sh

chmod +x install-version-os.sh
./install-version-os.sh --user


安装程序会将bazel安装到$HOME/bin目录下,需要把这个目录加入PATH

$ export PATH="$PATH:$HOME/bin"


安装其他依赖

apt-get install python-numpy swig python-dev


使用pip编译并安装

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


这行命令要一个相对目录,且:后面的需要跟该目录下的BUILD文件中的sh_binary的name字段保持一致

mkdir /tmp/tensorflow_pkg
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install /tmp/tensorflow_pkg/tensorflow-0.5.0-py2-none-any.whl

install时需要联网,如果联网失败会报错。至此,安装完毕。