鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

移植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”目录下: