tensorrt_demos: onnx to trt failed.. "cannot get YoloLayer_TRT plugin creator"
Hi. I try to convert yolo to trt.
I succeed to convert .cfg -> .onnx but failed .onnx -> .trt with Error “cannot get YoloLayer_TRT plugin creator”.
What should I check now and what does mean that error message.?
I already make in plugins dir and get libyolo_layer.so lib.
About this issue
- Original URL
- State: closed
- Created 3 years ago
- Comments: 29 (12 by maintainers)
I had the same issue of “RuntimeError: cannot get YoloLayer_TRT plugin creator”, because the environment CUDA_VISIBLE_DEVICES was set to -1. after export CUDA_VISIBLE_DEVICES=0, it worked.