开发者
下载

TorchNPU

作为Ascend for PyTorch社区的核心组件,TorchNPU是昇腾专为PyTorch打造的深度学习适配框架,使PyTorch框架能够直接调用昇腾NPU,为开发者提供强劲的AI算力支撑。

TorchNPU master文档说明

本文档基于master开发分支生成,当前处在活跃开发状态(Active Development State)并非稳定版本的正式发布文档。本文档内容具有时效性且处于持续更新中,此处描述的规格与行为可能随开发进度调整,不代表最终发布版本。如需稳定使用请使用最新正式发布的版本,请参见26.1.0版本文档

重要提示:

  • API不稳定:本文档所描述的接口、算子功能及性能参数可能随代码提交发生变动,且不保证向前兼容。
  • 功能差异:当前实现可能包含实验性特性,这些特性尚未经过完整的大规模稳定性及鲁棒性验证。
  • 安装方式差异
    • 正式发布版本:提供快速安装及源码编译等多种安装方式。
    • master开发分支:master在持续更新迭代中,因此未提供软件包形式的快速安装方式,请使用源码编译进行安装操作。
    • PyTorch 2.13.0即时适配版本:为跟随PyTorch社区的发布节奏,提供针对PyTorch 2.13.0版本的即时适配版本,可前往快速安装页面获取。该版本是基于master分支的较早快照,文档中部分新特性可能暂不可用,具体支持情况请参考Release
  • CANN配套关系:CANN基础依赖请使用最新正式版本,若需体验新特性或进行开发,可使用最新的CANN Weekly 版本
  • 使用建议:建议生产环境用户请移步查阅最新26.1.0版本文档
  • 反馈贡献:如果您在基于此开发版测试时发现Bug或文档歧义,欢迎通过Issue或PR提交反馈,共同推进TorchNPU的演进。

快速上手

TorchNPU概述