同步操作将从 Ascend/op-plugin 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
本项目开发了NPU PyTorch算子插件,为使用PyTorch框架的开发者提供便捷的NPU算子库调用能力。 OP-Plugin算子插件的编译、使用依赖昇腾PyTorch Adapter。因此,在编译op_plugin之前,需要了解、安装昇腾PyTorch。使用手册可参考昇腾社区pytorch。
暂未正式发布
安装依赖
安装时需要安装系统依赖及官方PyTorch框架。安装指导可参考昇腾pytorch社区安装依赖。
配置CANN环境变量脚本。
source <CANN软件安装目录>/<CANN软件路径>/set_env.sh
环境变量脚本的默认路径一般为:/usr/local/Ascend/ascend-toolkit/set_env.sh,其中ascend-toolkit路径取决于安装的CANN软件名称。
编译生成插件的二进制安装包。
下载对应op_plugin版本分支代码,进入插件根目录,以master为例
git clone https://gitee.com/ascend/op-plugin.git
cd op-plugin
可指定Python版本编包。如,--python3.8(缺省) 或 --python3.9 可指定昇腾pytorch版本编包。当前支持昇腾pytorch 1.11/2.0/2.1(master)版本。如,--pytorch=master(缺省) 或 --pytorch=v1.11.0 或 --pytorch=v2.0.1。
bash ci/build.sh --python=3.8 --pytorch=master
编译过程中,会在插件根目录新建build文件夹,并下载昇腾pytorch对应版本的源码,协同编译。 若build/pytorch目录存在,则编译op-plugin时,不再重复下载昇腾pytorch源码。如需下载所依赖的最新昇腾pytorch源码,删除build/pytorch目录即可。
完成编译后,安装dist目录下生成的插件torch_npu包,如果使用非root用户安装,需要在命令后加**--user**。
pip3 install --upgrade dist/torch_npu-1.11.0-cp37-cp37m-linux_aarch64.whl
# 若用户在x86架构下安装插件,请替换为对应的whl包。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。