tensorflow_addons提供了tensorflow2核心库之外的功能API,比如说tensorflow1.7版本中的某写功能在TensorFlow2中确实,就有可能在tensorflow_addons中找到。
在ARM64架构上tensorflow_addons不能直接通过pip、conda等工具安装,需要从源码进行编译安装。博主分享自己在鲲鹏(AMR64)上源码安装tensorflow_addons的经验给大家。
- 前期准备
开发环境:ARM64(鲲鹏上),centos7操作系统
安装 python\bazel\tensorflow - 编译安装
坑来: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
即可。 - 测试成功