win10 tensorrt源码编译onnx

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

win10 tensorrt源码编译onnx_第1张图片

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

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

win10 tensorrt源码编译onnx_第2张图片

编译成功后如下图:

win10 tensorrt源码编译onnx_第3张图片

你可能感兴趣的:(AI科普,人工智能)