[深度学习][onnxruntime]onnxruntime在windows10编译

Posted FL1623863129

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[深度学习][onnxruntime]onnxruntime在windows10编译相关的知识,希望对你有一定的参考价值。

测试环境:

windows10

cuda11.1.1

cudnn8.2.0

tensorrt8.2.3.0

VS2019

onnxruntime==1.12.1

编译流程:

git clone  --recursive https://github.com/Microsoft/onnxruntime

打开:x64 Native Tools Command Prompt for VS 2019

cd /d [youronnxrumtimepath]

.\\build.bat  --build_shared_lib --parallel --use_cuda --cuda_version 11.1 --cuda_home "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1" --cudnn_home "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1"  --build_csharp --use_tensorrt --tensorrt_home "D:\\lufiles\\TensorRT-8.2.3.0" --cmake_generator "Visual Studio 16 2019" --skip_tests --config Release

以上是关于[深度学习][onnxruntime]onnxruntime在windows10编译的主要内容,如果未能解决你的问题,请参考以下文章

[深度学习][环境配置]关于windows上如何使用cuda10.2支持onnxruntime的python版本

极智AI | 教你使用深度学习模型调试器 polygraphy

mask rcnn onnxruntime c++ 学习笔记

mask rcnn onnxruntime c++ 学习笔记

手把手教你使用OpenCV,ONNXRuntime部署yolov5旋转目标检测

无标题