Win11 跑通tensorRT

cuda_12.3.1_546.12_windows
TensorRT-8.6.1.6.Windows10.x86_64.cuda-12.0
cudnn-windows-x86_64-8.9.7.29_cuda12-archive

准备

1.安装cuda，成功之后文件夹如下图所示

2.下载cudnn，把cudnn对应的文件放在cuda里面

3.安装vs
4.安装对应cuda版本的tensorRT

https://developer.nvidia.com/tensorrt-download

5.opencv安装
编译好

打开vs，配置环境

用vs打开tensorRT中sample这个demo，双击sample_onnx_mnist.sln

打开项目选项，属性，添加include

添加lib

直接运行

遇到的问题：

解决：
debug就debug里面添加，release就在release里面添加

tensorRT的sample中，附加依赖项是这样的

kernel32.lib
user32.lib
gdi32.lib
winspool.lib
comdlg32.lib
advapi32.lib
shell32.lib
ole32.lib
oleaut32.lib
uuid.lib
odbc32.lib
odbccp32.lib
%(AdditionalDependencies)
nvinfer.lib
nvinfer_plugin.lib
nvonnxparser.lib
nvparsers.lib
cudnn.lib
cublas.lib
cudart.lib

2.需要开启很大的数组

3.量化
对onnx模型，进行量化

使用tensorRT
trtexec --onnx=yolov5s.onnx --saveEngine=model_int8.engine --int8 --fp16