[深度学习][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++ 学习笔记