centos7 arm服务器编译安装onnxruntime-gpu

11 篇文章 13 订阅 ¥29.90 ¥99.00
本文介绍了如何在CentOS7的ARM服务器上编译安装ONNXRuntime的GPU版本。由于官方仅提供CPU版,因此需要自行编译。编译前需准备Python3.8、CMake 2.26.0+、GCC 9.2以下版本。编译过程包括安装CUDA和CuDNN,然后下载ONNXRuntime源码并进行编译。最终编译完成的文件位于./build/Linux/Release/dist/目录下。
摘要由CSDN通过智能技术生成

前言

        ONNX Runtime是适用于Linux,Windows和Mac上ONNX格式的机器学习模型的高性能推理引擎,但在arm服务器上,onnxruntime只有CPU版的,GPU版的没有,因此需要自行去编译GPU版本的才可以。

环境准备

1、python3.8

2、cmake:2.26.0版本以上,可以直接下载aarch64版本的进行安装Releases · Kitware/CMake (github.com)

3、gcc:需要9.2版本以下,可以查看该篇文章进行编译【centos7 arm服务器编译安装gcc 9.2

4、onnxruntime源码:

要在CentOS 8上安装ONNX Runtime (ORT),你需要遵循一些步骤。ONNX Runtime是一个开源库,用于运行机器学习模型。以下是安装的基本流程: 1. **更新包管理器**: ```bash sudo yum update -y ``` 2. **安装依赖**: ORT需要Python的基础环境,所以首先安装Python3(如果尚未安装): ```bash sudo yum install epel-release -y sudo yum install python3 -y ``` 然后安装pip3,它是Python的包管理器: ```bash sudo yum install python3-pip -y ``` 3. **安装pip3额外的工具**: ```bash pip3 install wheel ``` 4. **下载并安装ONNX Runtime**: ONNX Runtime有官方的Python包,可以从其GitHub仓库获取: ```bash git clone https://github.com/microsoft/onnxruntime.git cd onnxruntime ``` 完成克隆后,导航到`cmake`文件夹并创建一个构建目录: ```bash mkdir build cd build ``` 5. **配置编译选项**: 创建一个CMakeLists.txt文件,如果已存在则打开编辑器(例如nano): ```bash nano CMakeLists.txt ``` 添加以下内容,配置为使用CPU版本,并启用Python接口(更多选项可以参考文档): ```cmake cmake .. \ -DCMAKE_BUILD_TYPE=Release \ -DPYTHON_EXECUTABLE=$(which python3) \ -DONNXRUNTIME_USE_MSVC_RUNTIME_LIBRARY_DLL=False \ -DUSE_CUDA=OFF \ -DUSE_OPENVINO=OFF \ -DUSE_MKLDNN=OFF \ -DUSE_TENSORRT=OFF \ -DUSE_NCCL=OFF \ -DUSE_ROCM=OFF \ -A x64 ``` 按`Ctrl+X`保存并退出。 6. **编译安装**: ```bash make -j$(nproc) sudo make install ``` 7. **检查安装**: 最后,你可以通过Python测试ONNX Runtime是否成功安装: ```bash python3 -c "import onnxruntime; print(onnxruntime.__version__)" ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

番茄小能手

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

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

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

打赏作者

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

抵扣说明:

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

余额充值