直接利用官方源碼,如下圖,trtexec源碼在TensorRT安裝目錄下,雙擊trtexec.sln文件,使用vs2019打開源碼工程。
如下圖,以yolov8為例子,編譯成功項目之后,設置命令行參數:
--onnx=d:/yolov8n.onnx --saveEngine=d:/yolov8n.trt --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640
編譯成功后如下圖: