赞
踩
win10电脑环境:
环境安装成功后,使用sampleOnnxMNIST测试
用vs打开sampleOnnxMNIST项目,位置在 D:\TensorRT-8.4.1.5\samples\sampleOnnxMNIST
右键项目>属性>VC++目录>包含目录

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
D:\TensorRT-8.4.1.5\include
D:\opencv45\opencv\build
D:\opencv45\opencv\build\include
右键项目>属性>VC++目录>库目录

D:\TensorRT-8.4.1.5\lib
D:\opencv45\opencv\build\x64\vc15\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib\x64
将TensorRT的链接库、OpenCV的链接库、CUDA的链接库分别配置到VS的附加依赖项中,如下图所示。
右键项目>属性>链接器>输入>附加依赖项

具体如下:
cublas.lib cublasLt.lib cuda.lib cudadevrt.lib cudart.lib cudart_static.lib cufft.lib cufftw.lib curand.lib cusolver.lib cusolverMg.lib cusparse.lib nppc.lib nppial.lib nppicc.lib nppidei.lib nppif.lib nppig.lib nppim.lib nppist.lib nppisu.lib nppitc.lib npps.lib nvblas.lib nvjpeg.lib nvml.lib nvrtc.lib OpenCL.lib nvparsers.lib nvonnxparser.lib nvinfer_plugin.lib nvinfer.lib opencv_world450.lib
如果是debug模式,最后一个就写成:opencv_world450d.lib
在预处理定义中添加:NOMINMAX和_CRT_SECURE_NO_WARNINGS

至此环境配置完成。
右击项目“重新生成”,重新编译项目,生成sample_onnx_mnist.exe
cmd 进入路径“D:\TensorRT-8.4.1.5\samples\sampleOnnxMNIST\x64\debug”
运行命令:sample_onnx_mnist.exe --datadir=D:\TensorRT-8.4.1.5\data\mnist\
即可测试成功,会出现下面的预测结果:

成功!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。