README.md
ONNX Runtime
项目配置
1、作者开发环境: Windows10 (vs 2019 or vs 2022)
2、项目依赖
- opencv 3.4.15
安装方法:自行安装配置项目依赖。 - 如模型运行平台 GPU :
- onnxruntime-win-x64-gpu-1.3.0
- cuda_10.1.105_418.96_win10
-
cudnn-10.1-windows10-x64-v7.6.5.32
安装方法:自行安装配置项目依赖。
- 如模型运行平台 CPU :
-
Microsoft.ML.OnnxRuntime.1.10.0 & Microsoft.ML.OnnxRuntime.MKLML.1.6.0
安装方法:
进入 visual studio 编辑环境,选择:工具 -> NuGet 包管理器 -> 管理解决方案的 NuGet程序包,
然后选择“浏览”,在搜索栏分别输入 “Microsoft.ML.OnnxRuntime.1.10.0”、“Microsoft.ML.OnnxRuntime.MKLML.1.6.0”,进行安装。
项目依赖工具包
ONNX 模型
项目使用方法
1、如GPU运行自行确定软件已安装及硬件环境符合。
- 注意:另外通过 main.cpp 中定义 USE_CUDA ,确定是运行在GPU or CPU 硬件上。