onnxruntime编译安装

最新推荐文章于 2025-05-12 20:54:02 发布

原创最新推荐文章于 2025-05-12 20:54:02 发布 · 858 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#人工智能

计算机视觉专栏收录该内容

5 篇文章

订阅专栏

本文介绍了如何从GitHub克隆Microsoft的onnxruntime库，切换到特定版本，使用CUDA进行编译，并在Qt项目中添加头文件和库链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

克隆源码

git clone --recursive https://github.com/Microsoft/onnxruntime
cd onnxruntime/
git checkout v1.8.0
git submodule update --init --recursive

编译

./build.sh --skip_tests --use_cuda --config Release --build_shared_lib --parallel --cuda_home /usr/local/cuda-11.1 --cudnn_home /usr/local/cuda-11.1

其中的use_cuda表示你要使用CUDA的onnxruntime，cuda_home和cudnn_home均指向你的CUDA安装目录即可。

编译完成输出

[100%] Linking CXX executable onnxruntime_test_all
[100%] Built target onnxruntime_test_all
[100%] Linking CUDA shared module libonnxruntime_providers_cuda.so
[100%] Built target onnxruntime_providers_cuda
2022-03-15 13:49:03,260 util.run [DEBUG] - Subprocess completed. Return code: 0
2022-03-15 13:49:03,260 build [INFO] - Build complete

qt添加头文件和库

*.pro文件

#onnx
INCLUDEPATH += /home/online/mytest/onnx/onnxruntime/include/onnxruntime \
            /home/online/mytest/onnx/onnxruntime/include/onnxruntime/core/session


LIBS += /home/online/mytest/onnx/onnxruntime/build/Linux/Release/libonnxruntime.so.1.8.0
LIBS += /home/online/mytest/onnx/onnxruntime/build/Linux/Release/libonnxruntime_providers_cuda.so
LIBS += /home/online/mytest/onnx/onnxruntime/build/Linux/Release/libonnxruntime_providers_shared.so