开发者
下载

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文档

表1 安装指南

安装指南

快速安装

软件安装