算子适配onnx,pt转onnx但报错。
收藏回复举报
算子适配onnx,pt转onnx但报错。
t('forum.solved') 已解决
发表于2024-06-17 17:51:46
0 查看

CANN:8.0.RC2.alpha001

硬件:atlas200 A

报错:torch.onnx.errors.UnsupportedOperatorError: ONNX export failed on an operator with unrecognized namespace npu::npu_add_custom. If you are trying to export a custom operator, make sure you registered it with the right domain and version.

我预先将addcustom包部署好,并适配了pytorch框架,测试可用;接着按照教程,在framework下新增了onnx_plugin,接着重新编译部署,于是生成onnx报错

我要发帖子