直接利用官方源码,如下图,trtexec源码在TensorRT安装目录下,双击trtexec.sln文件,使用vs2019打开源码工程。
如下图,以yolov8为例子,编译成功项目之后,设置命令行参数:
--onnx=d:/yolov8n.onnx --saveEngine=d:/yolov8n.trt --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640
编译成功后如下图: