独立研究者单块 GH200 预训练 21B MoE:MFU 从 21% 提到 50%

Prestigious-Taste-63 · reddit · 2026-10-12

一位独立研究者因预算受限,把原计划跑在 64 块 H100 上的 21B MoE(总参 20.99B、激活 3.36B,30 层,64 选 4 专家)预训练迁到单块 GH200(每小时 2.29 美元),并将全程测得的优化指标整理成论文公开。

关键结果:

踩坑记录:Transformer Engine FP8/行级 FP8 让 GEMM 变快但量化内核更慢;激活全卸载到 CPU 的分区方案;CPU 加法速度在相同设置下波动 22%–32%,最终用 C++ OpenMP 内核解决。

局限:只测了一种模型和一种硬件、各阶段指标混用代码版本非干净消融、未验证 FP8 对质量的影响、仅覆盖第一阶段约 11B tokens。代码、基准记录与训练日志已在 GitHub 开源,论文发布于 Zenodo。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →