讲师中心 微信公众号
AI工具推荐 视频效率加速

C++怎么使用ONNX Runtime推理AI模型

大墨小哥_5791

大墨小哥_5791

发布时间:2026-10-23 11:24:31

|

999人浏览过

|

来源于php中文网

原创

ONNX Runtime C++ API 初始化失败主因是动态库未正确链接或运行时找不到;需显式链接、配置环境变量、合理设置SessionOptions(含优化级、线程数、GPU支持);输入输出张量须严格匹配模型签名,注意内存布局与类型;推理时避免频繁小batch调用,推荐零拷贝与预分配。

c++怎么使用onnx runtime推理ai模型

ONNX Runtime C++ API 初始化失败:环境与链接问题

直接跑不起来,八成是环境没配对或链接错了。ONNX Runtime 的 C++ 接口依赖动态库(onnxruntime.dll 或 libonnxruntime.so),不是头文件包含完就完事的。

常见错误现象:undefined reference to `Ort::Env::Env(...)`(链接缺失)、Failed to load library: ...(运行时找不到 so/dll)、ORT_NO_SUCHFILE(模型路径错或格式非 ONNX)。

  • Linux/macOS 下必须显式链接 -lonnxruntime,且确保 LD_LIBRARY_PATH(或 DYLD_LIBRARY_PATH)包含库所在目录
  • Windows 下需把 onnxruntime.dll 放到可执行文件同目录,或系统 PATH 中;链接时加 onnxruntime.lib(不是 dll)
  • CMake 中推荐用 find_package(onnxruntime REQUIRED) + target_link_libraries(... ${ONNXRUNTIME_LIBRARIES}),避免手写路径出错

加载模型和创建会话:别漏掉 Ort::SessionOptions 配置

默认构造的 Ort::SessionOptions 不启用优化、不设线程数、不支持 GPU——这会导致推理慢得离谱,甚至在某些硬件上 fallback 到 CPU 也不报错。

典型误操作:直接 Ort::Session(session_options) 加载模型,结果性能比 Python 版还差一倍。

立即学习“C++免费学习笔记(深入)”;

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
  • 启用图优化:session_options.SetGraphOptimizationLevel(ORT_ENABLE_ALL)
  • 控制线程:session_options.SetIntraOpNumThreads(4)(CPU 场景下建议设为物理核数)
  • GPU 推理需额外初始化 Ort::SessionOptions 并传入 OrtCUDAProviderOptions,否则仍走 CPU
  • 模型路径必须是绝对路径或相对于当前工作目录的正确路径;相对路径出错时 Ort::Session 构造会抛 std::exception,务必 try-catch

输入输出张量:内存布局和类型必须严格匹配模型签名

ONNX 模型导出时固定了 input/output 的 shape、data_type 和 layout(通常是 NCHW 或 NHWC)。C++ 端手动构造 Ort::Value 时,只要有一个维度错、类型错(比如该用 ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT 却传了 double),就会触发 InvalidArgument 错误,且提示极不友好。

调试关键点:先用 onnxruntime_test_all 或 Python 的 ort.InferenceSession(model).get_inputs() 查清模型期望的 name/shape/dtype。

  • 输入数据必须是连续内存(std::vector<float></float> 而非 vector of vector),且按模型要求的 layout 排列(图像类模型多为 NCHW,别直接 memcpy cv::Mat.data)
  • 创建 Ort::Value 用 Ort::Value::CreateTensor(...),传入的 memory_info 建议用 Ort::MemoryInfo::CreateCpu(...) 或对应 GPU info
  • 输出 tensor 的 shape 需预先分配足够空间;GetTensorMutableData() 返回的是裸指针,别让它被 vector 释放

推理调用与性能陷阱:同步阻塞、零拷贝与 batch 处理

session.Run(...) 是同步阻塞调用,但它的耗时不仅含计算,还含输入 copy-in、output copy-out。高频小 batch 推理时,频繁调用 Run 会让内存拷贝成为瓶颈。

容易被忽略的细节:默认情况下,ONNX Runtime 对输入输出做深拷贝;若数据已在 GPU 上,又没配好 provider,它可能先拷回 CPU 再送进 kernel。

  • 启用 zero-copy 输入:用 Ort::Value::CreateTensor(..., Ort::MemType::MemTypeCPUInput)(仅限 CPU)或 GPU 对应 memory type,并确保 lifetime 覆盖整个 Run 过程
  • batch 推理必须提前 reshape 输入 tensor(如模型输入是 [1,3,224,224],想推 4 张图就得改第一维为 4,并确保数据内存连续排布)
  • 多线程并发推理需为每个线程创建独立 Ort::Session 实例(Ort::Env 可全局共享),否则可能触发内部锁竞争

模型输入 shape 动态变化、带 optional input、或用了 control flow op(如 Loop)时,C++ API 的 error 信息几乎不提供上下文,这时候得靠 onnx.checker.check_model() 和 onnx.shape_inference.infer_shapes() 在导出侧把模型验干净。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2248

2024.03.14

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

1019

2024.03.14

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

447

2024.03.14

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

327

2024.03.14

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

426

2024.03.14

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

580

2024.03.14

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

1409

2024.03.21

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

1217

2024.03.22

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

0

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++教程
C++教程

共115课时 | 43.3万人学习

C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn