返回全部动态
华为昇腾发布 TorchNPU:PyTorch 适配插件
原标题:Ascend/pytorch README.md
AI 摘要
华为昇腾社区发布 TorchNPU,这是面向昇腾 NPU 的 PyTorch 适配插件,使 PyTorch 框架可直接调用昇腾 NPU 算力。该插件复用上游 PyTorch 能力并针对昇腾 NPU 做深度适配,涵盖基础计算、分布式训练(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-09-11 15:59
抓取时间:2026-08-31 00:45
来源机构:Huawei Ascend