ubuntu 编译onnxruntime源码

14 篇文章 3 订阅
6 篇文章 1 订阅

编译成功截图:

环境:cuda11.0, cudnn8.0.5,tensorrt7.2.3, 显卡驱动是nvidia-driver-460

1.拉取源码 

# clone指定分支 发布版1.7.2
git clone -b rel-1.7.2 https://github.com/microsoft/onnxruntime.git

# 拉取子项目
git submodule update --init --recursive

拉取子项目非常耗时,可以点击下面的子项目链接,下载下来,再解压复制到对应的目录中(不建议这样子做,最好用上面命令)。

检查是否还有子类项目需要拉取:

git submodule update --init --recursive

没有输出,说明整个源代码拉取完毕。

2.编译

./build.sh --build_shared_lib --config Release --use_cuda --cudnn_home /usr/local/cuda/ --cuda_home /usr/local/cuda  --use_tensorrt --tensorrt_home /usr/lib/x86_64-linux-gnu/

出现的问题:

(1)CMake 3.13 or higher is required,cmake版本过低,不需要卸载cmake,直接更新即可

(2)onnx-tensorrt/builtin_op_importers.cpp:766:12: error: ‘class nvinfer1::IDeconvolutionLayer’ has no m,参考

这里编译onnxruntime,用到了cuda, cudnn, tensorrt,安装方法参考我的对应博文即可。注意如果tensorrt是deb方式安装,则cuda和必须要deb安装。

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Mr.Q

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值