文档中心TorchNPU返回顶部 TorchNPU 作为Ascend for PyTorch社区的核心组件,TorchNPU是昇腾专为PyTorch打造的深度学习适配框架,使PyTorch框架能够直接调用昇腾NPU,为开发者提供强劲的AI算力支撑。 TorchNPU 26.2.X的关键特性支持alltoallvc通信算子,优化alltoall通信性能。支持通信域以分级建链方式初始化,性能优化。支持Inductor基于AscendC IR的自动算子融合。继承DeviceAllocator,支持在Eager模式下启用accelerator memory和graph相关接口。从本版本起,Inductor后端中的Torch-MLIR编译器已废弃,涉及《Torch.compile》和《环境变量》,具体变更内容请参考版本配套文档。从本版本起,基于毕昇编译器提供的编译优化功能已废弃,可以使用Clang进行替换。 快速上手TorchNPU概述 版本说明 安装 查阅TorchNPU文档 表1 安装指南安装指南 安装 卸载与升级 参考 表2 用户指南用户指南 TorchNPU概述 快速入门 Torch.compile 配套软件库 故障处理 表3 APIAPI 原生API 自定义API 环境变量 表4 开发指南开发指南 内存管理 分布式 算子下发 故障诊断 NPUGraph 自定义算子适配开发 TorchAir 模型迁移调试 模型性能调优 表5 参考参考 昇腾产品形态说明 安全声明 术语