tensorrtx
tensorrtx copied to clipboard
使用程序将wts转成engine时报错,但是engine可以生成也可以使用
Env
- RTX2070
- Win10
- CUDA 11.6
- TensorRT 8.6.1.6
About this repo
- yolov5-v6.0
Your problem
我在使用程序将wts转成engine时报错,但是engine可以生成也可以使用;报错详细信息如下: (个人感觉这个报错只是提示回收方式不得当,但是对模型本身精度转换没有影响,因此这行ERROR是在转换完成后才提示的。想请教下大佬这个错误是什么原因造成的,希望能学习一下) [11/13/2023-19:06:28] [E] [TRT] 3: [builder.cpp::nvinfer1::builder::Builder::~Builder::341] Error Code 3: API Usage Error (Parameter check failed at: builder.cpp::nvinfer1::builder::Builder::~Builder::341, condition: mObjectCounter.use_count() == 1. Destroying a builder object before destroying objects it created leads to undefined behavior.)
最新的7.0已经修了这个问题,可以试下
最新的7.0已经修了这个问题,可以试下
好的,十分感谢大佬,我去试试
最新的7.0已经修了这个问题,可以试下
你好,yolov7也存在相同的问题
@nantututu 试下yolov5-7.0,不是yolov7
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
@Zhang-Wen-chao yolov7,把 delete builder放到delete engine之后试试,如果ok帮忙提个pr吧。
https://github.com/wang-xinyu/tensorrtx/blob/62050de3e7a6a7d88d93eb40dbea7cef2df777ac/yolov7/main.cpp#L47
@Zhang-Wen-chao yolov7,把 delete builder放到delete engine之后试试,如果ok帮忙提个pr吧。
https://github.com/wang-xinyu/tensorrtx/blob/62050de3e7a6a7d88d93eb40dbea7cef2df777ac/yolov7/main.cpp#L47
It works. 我已经提交了pr。
我之前改成下面这样,也work了,原来是歪打正着。
config->destroy(); serialized_engine->destroy(); builder->destroy();