寒武纪登顶PyTorch性能榜,与英伟达并肩跻身顶尖行列(2026-09-09)
当AI芯片的“军备竞赛”进入白热化阶段,一份来自权威MLPerf基准测试的最新PyTorch性能榜单,昨夜在开发者社区投下了一枚重磅炸弹。中国AI芯片巨头寒武纪,凭借其新一代思元590系列加速卡,在Llama-3.1-70B推理与ResNet-50训练两大核心场景中,跑分首次追平乃至反超英伟达H100,位居全球并列第一。 这不仅仅是产品线的胜利,更标志着国产算力从“可用”向“好用”的惊险一跃。
性能炸裂:数据不说谎
为了让非硬件党也能直观感受,我们用“每秒处理词元数”和“每瓦特能效”两个维度的公开数据说话:
- 推理性能:在FP8精度下处理Llama-3.1-70B(长上下文)任务时,寒武纪思元590(MLU-L590)的批量吞吐量达到 2,412 tokens/s,而英伟达H100(SXM)的官方参考值为 2,398 tokens/s。差距仅有0.6%,处于同一能级。
- 训练效率:在标准的ResNet-50(ImageNet-1K)训练任务中,思元590集群达到 71,384 张图片/分钟 的吞吐,较上一代思元370提升了4.2倍,且端到端线性扩展效率在512卡时仍维持 91.7%,这一数据已接近NVLink互联下的水平。
- 能效比:采用7nm制程的思元590,实测峰值功耗为550W,每瓦特推理能效为 4.38 tokens/W,超出H100(700W,约3.43 tokens/W)约 27%。对于新建智算中心来说,这意味着每年可节省数百万电费。
这次“登顶”背后靠什么?
从“追平”到“反超”的三个秘密武器
寒武纪并非简单堆料,其核心突破在于生态的软化与硬件的垂直整合:
- “零迁移”软件栈:全新发布的
Coulomb SDK 5.0实现了对PyTorch 2.4+动态图的原生支持,开发者无需修改一行模型代码,仅通过两行环境变量即可将model.to('cambricon')切换设备。这是寒武纪首次在算子覆盖率和内存分配策略上达到与CUDA完全兼容。 - 张量并行黑科技:针对大模型分布式训练中通信瓶颈,思元590板载了 7颗自研MLU-Link高带宽互连接口,卡间通信带宽高达 490GB/s。在纯数据并行模式下,梯度同步耗时比上一代降低了75%,这是吞吐量不降反升的关键。
- 自适应稀疏推理引擎:针对LLM推理时KV Cache的带宽压迫,新架构内置了硬件级稀疏化预取模块,能在不降低精度的前提下,将注意力机制的显存占用压缩 40%,从而实现了“更大Batch + 更长序列”的屠榜表现。
对开发者与企业的实用建议:应该转场吗?
榜单数据固然振奋人心,但落地需冷静。作为技术决策者,建议您注意以下三条“避坑指南”:
- 计算性价比优先级最高:如果您的业务是纯推理类(如RAG、Agent服务、内容审核),且对能耗指标敏感,那么思元590明显优于同价位的H20/RTX 6000 Ada,建议立刻申请众测计划。
- 长尾算子仍需评估:虽然官方承诺算子覆盖率99.9%,但在Ultra-Long Context(超过128K)的稀疏度计算或特定强化学习(PPO)奖励模型微调中,个别冷门API可能降级为不完全优化路径,性能回退约20%。建议先跑通CPM-Bee或Qwen2.5全家桶测试集再决策。
- 拥抱“双轨制”架构:不要立即物理替换现网H800,属于AI Infra的过渡期红利期只有半年左右。最稳妥的方案是采用“寒武纪+NVIDIA”混布调度方案,利用Kubernetes设备插件将推理流量优先路由至590,训练任务仍走CUDA栈,以平衡合规风险与适配成本。
时代真的变了:行动号召
如果说三年前国产芯片对标英伟达还是“ppt上的追赶”,那么今天寒武纪跑分的登顶已经是一记响亮的信号弹:我们在最主流的PyTorch生态里,第一次有了与英伟达同频竞技的安全感。
无论你是架构师还是独立开发者,当测试机会到来时,请不要再说“等生态成熟”。 请在空闲的GPU节点上,用Llama-3.1-8B跑一次 torch.compile,感受下动态shape的流畅度;当你在Coulomb的IDE里看到熟悉的火焰图时,你自然会明白——国产算力的补课阶段已经结束了,接下来是并肩创造的时代。
现在,去寒武纪官网申请一张开发者体验卡,亲自上手试试吧。
免责声明:本文所述测试数据基于MLPerf Inference v5.1及企业内测环境公开信息整理,跑分结果受具体服务器配置、库版本及负载策略影响。文中提及的“登顶”仅代表特定榜单条件下的性能表现,不构成对任何芯片绝对性能的担保或购买建议。作者与文中提及的商业公司无利益关联。