Tensorflow Adapter For Ascend

View English

Tensorflow Adapter For Ascend（简称TF_Adapter）致力于将昇腾AI处理器卓越的运算能力，便捷地提供给使用Tensorflow框架的开发者。 开发者只需安装TF_Adapter插件，并在现有Tensorflow脚本中添加少量配置，即可实现在昇腾AI处理器上加速自己的训练任务。

您可以通过阅读 TF_Adapter接口文档 获取更多使用细节。

安装

从源码安装

您可以从源代码构建 TF_Adapter 软件包并将其安装在昇腾AI处理器环境上。

TF_Adapter 插件与 Tensorflow 有严格的匹配关系，从源码构建前，您需要确保已经正确安装了 Tensorflow v1.15.0 版本 。

tfadapter也支持由源码编译，进行源码编译前，首先确保你有昇腾910 AI处理器的环境，同时系统满足以下要求：

• Linux OS
• GCC >= 7.3.0
• CMake >= 3.14.0
• SWIG

下载源码

git clone https://gitee.com/ascend/tensorflow.git
cd tensorflow

执行脚本生成安装包

chmod +x build.sh
./build.sh

脚本执行成功后，会在output目录生成tfadapter.tar压缩文件。

安装插件包

解压tfadapter.tar文件，生成npu_bridge-1.15.0-py3-none-any.whl, 然后使用 pip 安装 TF_Adapter 插件。

pip install npu_bridge-1.15.0-py3-none-any.whl

需要注意的是， 您应当保证安装路径与您编译时指定的 python 解释器搜索路径是一致的。

贡献

欢迎参与贡献。

社区版本规划

https://gitee.com/ascend/tensorflow/wikis/Home?sort_id=3076366

Release Notes

Release Notes请参考RELEASE.

License

Apache License 2.0