文档中心TorchNPU返回顶部 TorchNPU 作为Ascend for PyTorch社区的核心组件,TorchNPU(原名Ascend Extension for PyTorch)是昇腾专为PyTorch打造的深度学习适配框架,使PyTorch框架能够直接调用昇腾NPU,为开发者提供强劲的AI算力支撑。 TorchNPU 26.1.0的关键特性原Ascend Extension for PyTorch和torch_npu现已统一更名为TorchNPU,点击查看详情。新增支持Ascend 950DT产品。Eager模式使能DVM无图融合算子,点击查看详情。NPU上支持LibTorch Stable ABI,点击查看详情。关键计算类API支持异构输入。支持输入张量全轴动态场景下的算子编译。 快速上手TorchNPU概述 版本说明 快速安装 查阅TorchNPU文档 表1 安装指南安装指南 快速安装 软件安装 表2 用户指南用户指南 TorchNPU概述 快速入门 Torch.compile 配套软件库 故障处理 表3 APIAPI 原生API 自定义API 环境变量 表4 开发指南开发指南 框架特性 TorchAir 模型迁移调试 模型性能调优 表5 参考参考 昇腾产品形态说明 安全声明 术语