当前位置:   article > 正文

TensoRT8.4_cuda11.6 sampleOnnxMNIST运行生成_compiling: ../common/sampleengines.cpp linking: ..

compiling: ../common/sampleengines.cpp linking: ../../bin/sample_onnx_mnist_

1、版本信息

win10电脑环境:

  • TensorRT:8.4.1.5
  • CUDA: 11.6
  • VS: 2019

环境安装成功后,使用sampleOnnxMNIST测试

2、VS2019环境配置

用vs打开sampleOnnxMNIST项目,位置在 D:\TensorRT-8.4.1.5\samples\sampleOnnxMNIST

(1)包含目录

右键项目>属性>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

(2)库目录

右键项目>属性>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

 (3)链接器

将TensorRT的链接库、OpenCV的链接库、CUDA的链接库分别配置到VS的附加依赖项中,如下图所示。

右键项目>属性>链接器>输入>附加依赖项

具体如下:

  1. cublas.lib
  2. cublasLt.lib
  3. cuda.lib
  4. cudadevrt.lib
  5. cudart.lib
  6. cudart_static.lib
  7. cufft.lib
  8. cufftw.lib
  9. curand.lib
  10. cusolver.lib
  11. cusolverMg.lib
  12. cusparse.lib
  13. nppc.lib
  14. nppial.lib
  15. nppicc.lib
  16. nppidei.lib
  17. nppif.lib
  18. nppig.lib
  19. nppim.lib
  20. nppist.lib
  21. nppisu.lib
  22. nppitc.lib
  23. npps.lib
  24. nvblas.lib
  25. nvjpeg.lib
  26. nvml.lib
  27. nvrtc.lib
  28. OpenCL.lib
  29. nvparsers.lib
  30. nvonnxparser.lib
  31. nvinfer_plugin.lib
  32. nvinfer.lib
  33. opencv_world450.lib

如果是debug模式,最后一个就写成:opencv_world450d.lib

(4)预处理器

在预处理定义中添加:NOMINMAX和_CRT_SECURE_NO_WARNINGS

至此环境配置完成。

3、运行

右击项目“重新生成”,重新编译项目,生成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\

即可测试成功,会出现下面的预测结果:

成功!

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号