当前位置:   article > 正文

ONNX Runtime windows 编译指南_onnxruntime windows

onnxruntime windows

本指南将指导您如何编译 ONNX Runtime,包括准备工作、构建命令和常见配置。

1. 准备

确保您已安装:

  • CMake(3.13 或更高)
  • Visual Studio(2019 或更高,含 Visual Studio Build Tools)
  • Python(3.6 或更高,用于 ONNX Runtime 安装)

2. 构建

  1. 下载并安装onnxruntime的1.17.1版本。
  2. 下载并安装CUDA的12.2版本。
  3. 下载并安装cuDNN的8.9.2.26版本。
  4. 设置系统环境变量CUDA_HOME,使其指向CUDA安装路径。
  5. 设置系统环境变量CUDNN_HOME,使其指向cuDNN安装路径。
  6. 下载并安装Protocol Buffers的21.12版本可执行文件。
  7. 导航至onnxruntime的安装路径。
  8. 打开命令提示符/终端,执行以下命令以编译支持CUDA的onnxruntime库:
.\build.bat --build_shared_lib --config Release --use_xnnpack --use_cuda --cuda_version 12.2 --skip_tests --path_to_protoc_exe="D:\protoc-21.12-win64\bin\protoc.exe" --build_dir="./build"

注意:在执行命令时,请根据实际情况替换D:\protoc-21.12-win64\bin\protoc.exe为你的Protocol Buffers可执行文件的实际路径。

        命令行构建编译文件的过程中可能出现多次报错停止,有些是因为下载库失败,或着中断导致的停止,重新执行编译命令即可,必要事改变上网方式,加快库的下载速度

        需要下载的文件均在github开源

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/668992
推荐阅读
相关标签
  

闽ICP备14008679号