Asecend Tensor Compiler简称ATC,昇腾张量编译器,主要是将开源框架的网络模型或Ascend IR定义的单算子描述文件(json格式)转换为昇腾AI处理器支持的om格式

场景描述:ONNXRuntime调用CANN后端执行时,报了没有找到te_fusion的error,从EROOR来看这是没有找到tensor engine融合的模块.

 Ascend配套软件包安装后,如下几个包需要先卸载再安装:

 若这几个包未安装,则会出现ModuleNotFoundError : No module named "te_fusion"的错误;

上述几个包的路径有可能在/home/HwHiAiUser/Ascend/ascend-toolkit/latest/fwkacllib/lib64/下面,根据自己情况进行确定
我的whl包分别是

topi-0.4.0-py3-none-any.whl
te-0.4.0-py3-none-any.whl
hccl-0.1.0-py3-none-any.whl

安装上述三个whl包之后可以正常运行

参考链接:

Asecend Tensor Compiler编译失败,ModuleNotFoundError:No module named “te_fusion“_FADEMANJU的博客-CSDN博客MindSpore:环境问题案例_modulenotfounderror: no module named 'te_fusion_skytier的博客-CSDN博客

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐