昇腾成为首个获PyTorch官方支持的中国算力底座

来源:互联网 时间:2026-09-15

PyTorch Conference China 2026 大会上,昇腾公布了与 PyTorch 社区合作的最新进展:昇腾算力正式上线 PyTorch 官网,成为 PyTorch 官方支持的第一个中国算力底座,同时接入社区 CI/CD 体系。对国内开发者来说,这件事的实际意义比「又多了一个适配版本」要大得多。

原生支持和插件适配不是一回事。过去很长一段时间,主流开源框架对国内硬件的支持大多以第三方插件形式实现。插件这套做法能用,但短板很明显:框架版本一升级,插件要跟着改;社区的质量门禁管不到插件,出了问题只能找厂家;标准化测试覆盖不全,开发者拿到的行为跟官方文档里写的不总是一致。原生支持解决的是这层额外的适配成本。硬件被主流社区接纳为后端,框架的每一次版本迭代都要考虑它的兼容性,这比事后打补丁可靠。

据大会上披露的信息,双方的合作已经从硬件适配走到社区共建:昇腾完成社区全量特性与测试用例的硬件后端解耦改造,PyTorch 核心 API 实现对昇腾硬件 100% 支持,功能、精度、性能层面完成对齐;PyTorch 官网后端页面已经纳入昇腾入口,开发者可以通过一行命令完成环境安装。质量看护方面,昇腾正式接入 PyTorch 社区 CI/CD 体系,2000 多张昇腾算力卡进入社区,做 7×24 小时自动化测试,重点是版本迭代过程中的代码兼容性。

技术协同上,算子层面有面向昇腾的融合算子生成,兼容 Triton 编程能力;框架层面的分布式组件对 FSDP 等训练方案的优化也会向社区回贡;推理侧则向主流推理框架贡献调度与部署相关特性。

对开发者来说,最先降下来的是迁移成本。以前从 GPU 换到国产卡,要处理一大堆算子不支持、精度对不上的问题;原生后端接入之后,主体代码不需要为硬件改写法。升级期的兼容性也有了保障。硬件进了社区 CI,意味着框架发新版本之前就有对应的测试跑过,不会出现框架一升级、整个训练脚本跑不起来的情况。文档和社区答案会逐渐通用。硬件官方化之后,官方文档、issue 讨论、第三方教程的可复用程度都会提高,遇到问题不再只能在厂家自己的论坛里找。

该压测的还得压测。官方支持是入口,不是性能保证。API 100% 覆盖和具体模型的实际吞吐是两件事,换硬件之前该压测的还是要压测,尤其是长序列训练和大 batch 推理这两种负载。生态成熟度还在爬坡。算子覆盖度、issue 响应速度、周边工具链的完整性都需要时间积累,新出的模型架构适配通常会有延迟。版本也得对齐。国内硬件后端与框架版本是绑定的,装环境时按官方给的组合来,别自己混搭,混搭的报错往往很难定位。一个更宏观的意义是:国产算力与主流开源生态从「能用」走向「共建」,对做长期项目的团队来说,选型时的顾虑会少一项。

相关文章

标签:

A5创业网 版权所有