tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Published 20 hours ago • verified publisher icon wang-xinyu

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

Open silas-yy opened this issue 1 month ago • 1 comments

配置环境： 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

silas-yy avatar May 31 '24 03:05 silas-yy