Windows下从源码编译ONNX Runtime

前言:作者在做深度学习模型部署过程中,遇到一些算子问题,需要在ONNX Runtime平台上实现一些自定义算子,在此过程中,onnxruntime官方给的现成的库缺少一些必要文件,遂需要下载onnxruntime源码并进行编译。

一、环境说明

windows 10

cuda V11.5

cudnn 8.8.0

cmake 3.24.3(试过其他版本,均无法编译顺利完成)Index of /files/v3.24 (cmake.org)

visual studio 2019

python 3.9(电脑需要有python环境,编译本质上是运行的build.py)

onnxruntime 1.11.1

二、编译流程

# 打开git bash
git clone -b rel-1.11.1 --recursive https://github.com/Microsoft/onnxruntime
cd onnxruntime
git submodule update --init --recursive

# 管理员打开vs的工具x64 Native Tools Command Prompt for VS 2019
d:
cd onnxruntime

执行脚本编译

 .\build.bat  --build_shared_lib --parallel --use_cuda --cuda_version 11.4 -
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值