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. 测试成功
    在这里插入图片描述
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值