AMR64(鲲鹏)上安装tensorflow_addons

tensorflow_addons提供了tensorflow2核心库之外的功能API,比如说tensorflow1.7版本中的某写功能在TensorFlow2中确实,就有可能在tensorflow_addons中找到。
在ARM64架构上tensorflow_addons不能直接通过pip、conda等工具安装,需要从源码进行编译安装。博主分享自己在鲲鹏(AMR64)上源码安装tensorflow_addons的经验给大家。

  1. 前期准备
    开发环境:ARM64(鲲鹏上),centos7操作系统
    安装 python\bazel\tensorflow
  2. 编译安装
    git clone https://github.com/tensorflow/addons.git
    cd addons
    
    # This script links project with TensorFlow dependency
    python3 ./configure.py
    
    bazel build build_pip_pkg
    bazel-bin/build_pip_pkg artifacts
    
    pip install artifacts/tensorflow_addons-*.whl
    
    坑来:
    运行 bazel build build_pip_pkg 出现如下报错信息:gcc: error: unrecognized command line option '-mavx',这是因为configure.py脚本里面默认有 -mavx 选项,而这个选项并不是必须的,可以直接在configure.py中去掉这行代码。再重新运行 python3 ./configure.py 即可。
  3. 测试成功
    AMR64(鲲鹏)上安装tensorflow_addons_第1张图片

你可能感兴趣的:(技术,tensorflow,python)