前言
- 笔者搭建时,很多文章已经很老,版本对应比较久远了。这里采用的是新版
- 版本对应很重要,尝了一下onnxruntime-gpu最新版1.60搭配cuda1.7
- 文末给测试搭建是否成功的简易代码。
环境版本
采用的是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;
}