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博客