win10 tensorrt源码编译onnx

直接利用官方源码,如下图,trtexec源码在TensorRT安装目录下,双击trtexec.sln文件,使用vs2019打开源码工程。

如下图,以yolov8为例子,编译成功项目之后,设置命令行参数:

--onnx=d:/yolov8n.onnx  --saveEngine=d:/yolov8n.trt  --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640

编译成功后如下图:

猜你喜欢

转载自blog.csdn.net/m0_72734364/article/details/134599859