Skip to content

IPluginCreator not found in Plugin Registry error when deserializing engine model #1913

@huanghaifeng1234

Description

@huanghaifeng1234

Description

I train yolov5 using pytorch, and convert it to tensorrt engine file. When converting I use REGISTER_TENSORRT_PLUGIN(YoloPluginCreator) to register my custom plugin. When loading the engine file, I try to deserialize it but failed, below is the error infomation:

[04/11/2022-20:45:35] [E] [TRT] 1: [pluginV2Runner.cpp::nvinfer1::rt::load::290] Error Code 1: Serialization (Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry)
[04/11/2022-20:45:35] [E] [TRT] 4: [runtime.cpp::nvinfer1::Runtime::deserializeCudaEngine::50] Error Code 4: Internal Error (Engine deserialization failed.)

I have added
initLibNvInferPlugins(&gLogger, ""); before serialization but that doesn't work for me.

Environment

TensorRT Version: 7.2.3.4
NVIDIA GPU: GeForce GTX 1660 Ti
NVIDIA Driver Version: 496.49
CUDA Version: 11.0
CUDNN Version: 8.1.1
Operating System: windows11
Python Version (if applicable):
Tensorflow Version (if applicable):
PyTorch Version (if applicable): 1.11.0
Baremetal or Container (if so, version):

Relevant Files

Steps To Reproduce

serialize:
image
image
image

deserialize:
image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions