前言

  1. 笔者搭建时,很多文章已经很老,版本对应比较久远了。这里采用的是新版
  2. 版本对应很重要,尝了一下onnxruntime-gpu最新版1.60搭配cuda1.7
  3. 文末给测试搭建是否成功的简易代码。

环境版本

采用的是vs2022、cuda1.7、cudnn8.9.1、onnxruntime-gpu1.60

onnxruntime下载

选择所需对应版本
请添加图片描述
点击绿色背景内链接跳转下载
请添加图片描述
下载完成后保存到熟悉路径

vs

创建新项目

选择 工具->NuGet管理包->程序包管理控制台,添加刚刚buget包到设置中的程序包源:
在这里插入图片描述
添加完成后点击浏览,右侧界面中安装配置。

完成后添加依赖

打开属性管理器,vc++zhong包含目录添加头文件路径:

build/native/include

请添加图片描述
继续添加算法库文件 :

runtime/win-x64

请添加图片描述
添加连接器中输入:
请添加图片描述
附加依赖项:
请添加图片描述

环境变量

将 onnxruntime.dll 所在的文件夹添加到系统环境变量。

测试是否成功

#include <iostream>
#include <onnxruntime_cxx_api.h>

int main(int argc, char* argv[]) {
 Ort::Env env(ORT_LOGGING_LEVEL_WARNING, "test");
 return 0;
 }
Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐