5

能够将 pytorch 模型转换为 onnx 并使用 pytorch 验证 onnx 输出。它工作正常。但是当我尝试将 onnx 转换为 tensorrt 引擎时,它在 onnx 解析器上失败并返回此错误。

[TensorRT] 错误:INVALID_ARGUMENT:getPluginCreator 找不到插件 ScatterND 版本 1

Onnx 版本:1.6.0 Tensorrt 版本:7.1.3.4

谁能在这里提供帮助,如何在 python 中添加 ScatterND 插件以及如何使用它?有没有其他方法可以解决这个问题?

提前致谢。

4

0 回答 0