如何解决将onnx转换为Tensorrt模型时找不到ScatterND插件
能够将pytorch模型转换为onnx并使用pytorch验证onnx输出。工作正常。 但是当我尝试将onnx转换为tensorrt引擎时,它在onnx解析器上失败并返回此错误。
[TensorRT]错误:INVALID_ARGUMENT:getPluginCreator找不到插件ScatterND版本1
Onnx版本:1.6.0 Tensorrt版本:7.1.3.4
任何人都可以在这里提供帮助,如何在python中添加ScatterND插件以及如何使用它? 还有其他方法可以解决此问题吗?
谢谢。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。