Alex' Blog

记录生活和工作

0%

Tensorflow2.0源码安装

安装步骤(mac)

基础环境

操作系统:Mac os

环境:Python3,JDK 8

安装编译工具

安装bazel;推荐使用homebrew安装

Note: 版本要求要低于0.26.1,所以在安装时候需要指定版本,选择合适的formula.

这里给出一个合适的地址:https://raw.githubusercontent.com/bazelbuild/homebrew-tap/c8a0ccc2a8b442d9887d88c6380f835f533ffd47/Formula/bazel.rb

1
brew install https://raw.githubusercontent.com/bazelbuild/homebrew-tap/c8a0ccc2a8b442d9887d88c6380f835f533ffd47/Formula/bazel.rb

下载源代码

1
2
git clone tensorflow
git checkout (branch you want to install)

安装相关依赖

1
2
3
pip install -U --user pip six numpy wheel setuptools mock 'future>=0.17.1'
pip install -U --user keras_applications==1.0.6 --no-deps
pip install -U --user keras_preprocessing==1.0.5 --no-deps

配置

1
./configure

这里大部分都选择默认的no,可以开启XLA优化

编译

1
bazel build //tensorflow/tools/pip_package:build_pip_package

接下来是漫长的等待,我的机器编译了一天(mbp)。

编译python包

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

得到一个whl包,通过pip安装即可将tensorflow安装至python环境。