返回全部动态

华为昇腾 TorchNPU:PyTorch 的昇腾 NPU 适配插件

原标题:Ascend/pytorch README.md

华为昇腾 Ascend PyTorch Activity一手来源开源质量 64

AI 摘要

华为昇腾社区发布 TorchNPU 项目 README,介绍这是为 PyTorch 框架打造的昇腾 NPU 适配插件,使 PyTorch 可直接调用昇腾 AI 处理器。文档列出基础算子、分布式训练(FSDP2、DTensor、集合通信)、图模式加速、调试调优及内存管理等核心模块,并给出安装方式与版本兼容信息。最新动态显示 TorchNPU 26.0.0 已支持 PyTorch 2.10.0 与 Python 3.13。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

<h1 align="center">TorchNPU</h1> <p align="center"> <strong>PyTorch Adapter Plugin for Ascend NPU</strong> </p> <p align="center"> English | <a href="./README.zh.md">简体中文</a> </p> <p align="center"> <a href="#supported-pytorch-versions">Supported Versions</a> · <a href="#installation">Installation</a> · <a href="#quick-start">Quick Start</a> · <a href="./COMPATIBILITY.en.md">Compatibility</a> · <a href="./SUPPORT.en.md">Support</a> · <a href="./CONTRIBUTING.en.md">Contributing


发布时间:2026-10-28 14:25
抓取时间:2026-08-31 00:45
来源机构:Huawei Ascend
阅读原文github.com