在模型运行时遇到报错“TVM/te/cce error.”

报错信息

问题分析

PyTorch内调用NPU类型算子时,强依赖于te、cce、TVM组件,PyTorch、CANN/nnae和te版本需要一致。在更新CANN/nnae后,te等组件不会自动更新,当版本不匹配时,则会出现该报错。

处理方法

更新te等组件版本,具体需要更新te-*.whl和topi-*.whl安装包。

在CANN安装目录或者nnae安装目录的lib64子目录下(以root安装用户为例:默认安装路径在/usr/local/Ascend/ascend-toolkit/latest/lib64目录下,更新安装包即可),找到te-*.whl和topi-*.whl安装包,使用pip3 install命令分别执行安装即可。