ubuntu20编译TensorFlow 源码

参考https://tensorflow.google.cn/install/source?hl=zh_cn

1. 安装依赖

2. 安装Bazel

从https://github.com/bazelbuild/bazel/releases?after=3.2.0下载了3.1.0的安装包https://github.com/bazelbuild/bazel/releases/download/3.1.0/bazel-3.1.0-installer-linux-x86_64.sh安装。

3. 下载 TensorFlow 源代码

git clone https://github.com/tensorflow/tensorflow.git

现在的最新master开发分支应该是TensorFlow 2.x版本的。

4. 配置build

cd tensorflow
./configure

5. 构建 pip 软件包

bazel build //tensorflow/tools/pip_package:build_pip_package --local_ram_resources=2048
./bazel-bin/tensorflow/tools/pip_package/build_pip_package --nightly_flag /tmp/tensorflow_pkg

6. 安装软件包

pip install /tmp/tensorflow_pkg/tf_nightly-2.5.0-cp38-cp38-linux_x86_64.whl

安装成功后，打印一下版本号：

python

>>> import tensorflow
>>> tensorflow.__version__
'2.5.0'
>>> exit()

觉得tensorflow很不好编译，以前编译总是失败，总算成功一次，也是晕晕乎乎的。bazel也觉得不好用，编译tensorflow时间比较长，如果中间出了问题，就又从头开始编译，盯了一天终于用虚拟机编译成功一次。

https://download.csdn.net/download/eidolon_foot/13799893

 

查询tensorflow安装路径为:

tensorflow.__path__

参考：https://blog.csdn.net/u011961856/article/details/76861052