atc-model-converter - 面向昇腾 NPU 的 PyTorch 或 ONNX 模型转换
分析模型代码,将 PyTorch 导出为 ONNX,经 ATC 转换为 OM,适配昇腾端到端推理并验证精度。
标签
更新于: 2026-10-19能力
典型输入
典型输出
该技能可以做什么
- 发现模型参数
- 探测运行时张量形状
- 导出 PyTorch 为 ONNX
- 检查 ONNX 输入输出
- 转换 ONNX 为 OM
- 适配端到端昇腾推理
- 对比 ONNX 与 OM 精度
- 生成可复现 README
输入
- 模型权重路径
- 模型代码仓库
- 任务类型
- 目标输入尺寸
- 测试图片或数据
- 目标昇腾 SoC 版本
输出
- 参数发现报告
- ONNX 模型文件
- OM 模型文件
- 昇腾推理流水线
- 精度对比结果
- 可复现 README
要求
- Python 3.10 或更低版本
- NumPy 低于 2.0
- CANN 8.1.RC1 或更高版本
- 昇腾 ATC 环境
- 支持 ONNX opset
