移植tensorflow 2.4.1

操作步骤

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 执行以下命令,创建tensorflow构建环境。

    conda create -n bazel_build bazel=3.7 python=3.7.12  -c conda-forge

  3. 执行以下命令,激活构建环境。

    conda activate bazel_build

  4. 执行以下命令,安装构建依赖。

    conda install -c conda-forge keras-preprocessing=1.1.2 numpy=1.18.5

  5. 执行以下命令,下载tensorflow2.4.1源码。

    git clone https://github.com/tensorflow/tensorflow.git -b v2.4.1

  6. 执行以下命令,进入源码目录。

    cd tensoflow

  7. 执行以下命令,配置build。

    python configure.py

  8. 执行以下命令,源码编译。

    bazel build //tensorflow/tools/pip_package:build_pip_package

    构建过程中会出现很多错误,因为网络问题,很多在资源bazel构建时不能自动获取,需要将所需资源下载下来,并将构建文件中的url修改为本地路径。具体参考编译tensorflow 2.4.1时,很多报错信息在资源bazel构建时不能自动获取

  9. 执行以下命令,构建tensorflow本地conda包。

    bazel-bin/tensorflow/tools/pip_package/build_pip_package /path/to/tensorflow

    构建好的tensorflow安装包存放在“/path/to/ tensorflow”目录下: