Skip to content

Ascend/tensorflow

Repository files navigation

Tensorflow Adapter For Ascend

View English

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

tfadapter

您可以通过阅读 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