ONNX安装与配置完全指南

最新推荐文章于 2024-09-13 22:19:58 发布

郦升卿Roberta

最新推荐文章于 2024-09-13 22:19:58 发布

阅读量262

点赞数 2

本文链接：https://blog.csdn.net/gitblog_09111/article/details/142223424

版权

ONNX安装与配置完全指南

onnx Open standard for machine learning interoperability 项目地址: https://gitcode.com/gh_mirrors/onn/onnx

项目基础介绍及主要编程语言

ONNX (Open Neural Network Exchange) 是一个开放的生态系统，旨在赋能AI开发者随着项目的进展自由选择合适的工具。它支持深度学习和传统机器学习模型，并定义了一个可扩展的计算图模型，包括内置运算符和标准数据类型定义。当前重点在于满足推理（评分）的需求。ONNX被广泛应用于多种框架、工具和硬件中，促进了不同框架间的互操作性，加速了从研究到生产的路径，促进了AI领域的创新。项目主要采用 Python 进行开发，同时也涉及C++等其他语言。

关键技术和框架

计算图模型: ONNX使用计算图来表示机器学习模型，这使得模型可以在不同的平台间移动。
模型标准化: 定义了一套统一的运算符集合，确保模型在不同实现之间的兼容性。
跨平台支持: 支持在各种运行环境如CPU、GPU和其他加速器上运行模型。
社区驱动: 基于开放治理结构，鼓励开发者参与贡献。

安装和配置步骤

准备工作

安装Git: 首先，确保你的系统上已安装Git。如果未安装，访问Git官网下载并安装。
Python环境: 确保安装有Python 3.6或更高版本，并建议通过Anaconda管理Python环境以简化依赖项管理。

安装ONNX

通过pip安装(推荐)

打开命令提示符或者终端。
安装最新稳定版ONNX:
```
pip install onnx
```
若要包括可选的参考实施依赖项，使用：
```
pip install onnx[reference]
```

从源代码编译安装

如果你希望从源码编译安装ONNX，以获得自定义配置或最新特性，按照以下步骤操作：

克隆项目源码:

git clone https://github.com/onnx/onnx.git
cd onnx

准备依赖: 确保你的系统上安装了C++17或更高版本的编译器。若未安装protobuf，ONNX会尝试自动下载并编译。或者手动安装protobuf库（及其C++组件），之后可能需要设置环境变量来指明是使用共享库还是静态库，例如：
```
export CMAKE_ARGS="-DONNX_USE_PROTOBUF_SHARED_LIBS=ON"  # 对于Linux，取决于protobuf的安装形式
set CMAKE_ARGS="-DONNX_USE_PROTOBUF_SHARED_LIBS=ON"   # 对于Windows
```
编译并安装: 在ONNX根目录下执行：
```
python setup.py install
```
或者使用pip进行局部安装（不修改全局Python环境）：
```
pip install .
```
验证安装: 安装完成后，你可以通过Python来检验ONNX是否正确安装：
```
import onnx
print(onnx.__version__)
```
如果成功显示版本号，则表明安装完成。