search

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API
MIT License
7.04k stars 1.78k forks source link

ubuntu下yolov5_v5.0的.wts转.engine失败 #1534

Closed silas-yy closed 2 days ago

silas-yy commented 6 months ago

配置环境： ubuntu 20.04 cuda v11.2 TensorRT 8.6.1

About this repo

cmake成功，make成功 yolov5_v5跑的，.pt文件也是从yolov5_v5中下载的yolov5s，.pt转.wts成功

Your problem

sudo ./yolov5 -s ../yolov5s.wts yolov5s.engine s报错： [05/31/2024-10:57:10] [E] [TRT] 2: [builder.cpp::createCaskKernelLibraryImpl::169] Error Code 2: Internal Error (Assertion validateCaskKLibSize(buffer.size) failed. ) 段错误 image 搜出来是版本不兼容，但我之前用trtexec，将onnx模型转为engine模型可以，且测试通过 image

wang-xinyu commented 5 months ago

这个报错应该是环境有问题，trt版本和依赖cuda等版本不匹配

stale[bot] commented 2 months ago

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.