search

Monday-Leo / YOLOv7_Tensorrt

A simple implementation of Tensorrt YOLOv7
108 stars 15 forks source link

执行infer.py推理没有结果，请大佬指点一下 #5

Open BandyKenny opened 2 years ago

BandyKenny commented 2 years ago

`D:\Anaconda3\envs\torch17\python.exe D:/work_place/yolov7_coil_test/infer_python.py [07/18/2022-18:04:34] [TRT] [I] [MemUsageChange] Init CUDA: CPU +578, GPU +0, now: CPU 8750, GPU 961 (MiB) [07/18/2022-18:04:34] [TRT] [I] Loaded engine size: 75 MiB [07/18/2022-18:04:34] [TRT] [W] TensorRT was linked against cuBLAS/cuBLAS LT 11.6.3 but loaded cuBLAS/cuBLAS LT 11.2.0 [07/18/2022-18:04:34] [TRT] [I] [MemUsageChange] Init cuBLAS/cuBLASLt: CPU +701, GPU +262, now: CPU 9535, GPU 1295 (MiB) [07/18/2022-18:04:35] [TRT] [I] [MemUsageChange] Init cuDNN: CPU +476, GPU +254, now: CPU 10011, GPU 1549 (MiB) [07/18/2022-18:04:35] [TRT] [W] TensorRT was linked against cuDNN 8.2.1 but loaded cuDNN 8.0.4 [07/18/2022-18:04:35] [TRT] [I] [MemUsageChange] TensorRT-managed allocation in engine deserialization: CPU +0, GPU +71, now: CPU 0, GPU 71 (MiB)

Process finished with exit code -1073741819 (0xC0000005) ` 我的转出来的onnx是： image

Monday-Leo commented 2 years ago

tensorrt版本需要高于8.2，并且从输出中看出cuda cudnn版本和tensorrt没有配合，请到英伟达官网看一下版本关系。

BandyKenny commented 2 years ago

image 我得Tensorrt版本是>8.2的，也可以正常生成.eng文件 image 但是进行infer.py脚本就会没反应： image 卡住后，最后得到的结果就是： image

Monday-Leo commented 2 years ago

请加Q群:768071513交流。

shuxp commented 2 years ago

我也遇到了这个问题，大佬们解决了吗？

shuxp commented 2 years ago

我也遇到了这个问题，大佬们解决了吗？

找到问题了，trt自身bug，trt8.4修复。