同步操作将从 Ascend/op-plugin 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
本项目开发了Ascend Extension for Pytorch(torch_npu)算子插件,为使用PyTorch框架的开发者提供便捷的NPU算子库调用能力。 OpPlugin算子插件的编译、使用依赖昇腾Ascend Extension for PyTorch。因此,在编译OpPlugin之前,需要了解、安装昇腾PyTorch。使用手册可参考昇腾社区Ascend Extension for Pytorch。
暂未正式发布
安装依赖
安装时需要安装系统依赖及官方PyTorch框架。安装指导可参考Ascend Extension for Pytorch。
配置CANN环境变量脚本。
source <CANN软件安装目录>/<CANN软件路径>/set_env.sh
环境变量脚本的默认路径一般为:/usr/local/Ascend/ascend-toolkit/set_env.sh,其中ascend-toolkit路径取决于安装的CANN软件名称。
编译生成插件的二进制安装包。
下载对应OpPlugin版本分支代码,进入插件根目录,以master为例
git clone https://gitee.com/ascend/op-plugin.git
cd op-plugin
执行编译构建,当前支持昇腾pytorch 1.11/2.0/2.1/2.2/2.3/master版本
bash ci/build.sh --python=3.8 --pytorch=v2.1.0
参数 | 取值范围 | 说明 | 缺省值 | 备注 |
---|---|---|---|---|
pytorch | v1.11.0, v2.1.0, v2.2.0, v2.3.0 master | 指定编译过程中使用的pytorch版本 | master | master对应的pytorch版本为2.4 |
python | 3.7, 3.8, 3.9, 3.10 | 指定编译过程中使用的python版本 | 3.8 | 仅pytorch版本为1.11时才支持指定python版本为3.7 |
编译过程中,会在插件根目录新建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包。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。