linghu8812 / tensorrt_inference

699 stars 205 forks source link

YOLOV5: getPluginCreator could not find plugin ScatterND version 1 #108

Open MichaelWU0726 opened 3 years ago

MichaelWU0726 commented 3 years ago

编译生成yolov5_trt可执行文件后,运行yolov5_trt报错,ScatterND算子不支持,要如何改写? (作者工程真的自己编译过吗。。。自己没pass一遍就pull仓库了吗?)

[TRT] ModelImporter.cpp:135: No importer registered for op: ScatterND. Attempting to import as plugin.
[TRT] builtin_op_importers.cpp:3771: Searching for plugin: ScatterND, plugin_version: 1, plugin_namespace:
[TRT] INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1
ERROR: builtin_op_importers.cpp:3773 In function importFallbackPluginImporter:
Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?"
roborocklsm commented 2 years ago

You need to update your TensorRT OSS version. Build one from git@github.com:NVIDIA/TensorRT.git

MichaelWU0726 commented 2 years ago

You need to update your TensorRT OSS version. Build one from git@github.com:NVIDIA/TensorRT.git

I used TensorRT OSS version7.2.3.4 . I found the op ScatterND is supported in version8.0 later, but the reposity was built before releasing version8. I think the project still work without updating to version8

roborocklsm commented 2 years ago

You need to update your TensorRT OSS version. Build one from git@github.com:NVIDIA/TensorRT.git

I used TensorRT OSS version7.2.3.4 . I found the op ScatterND is supported in version8.0 later, but the reposity was built before releasing version8. I think the project still work without updating to version8

有道理

MichaelWU0726 commented 2 years ago

You need to update your TensorRT OSS version. Build one from git@github.com:NVIDIA/TensorRT.git

I used TensorRT OSS version7.2.3.4 . I found the op ScatterND is supported in version8.0 later, but the reposity was built before releasing version8. I think the project still work without updating to version8

有道理

So i am confused how the owner build the project and pass.

linghu8812 commented 2 years ago

@MichaelWU0726 try with the latest code of linghu8812/yolov5