LLM Activation Checkpointing 生产实战:用 Selective Recompute、非重入模式与 RNG 一致性压低训练显存
大模型训练中,激活值常比参数更早耗尽显存。本文深入探讨选择性重计算、非重入模式、随机数状态一致性及工程门禁策略,帮助团队在保证梯度正确的前提下降低峰值显存,控制训练吞吐损失。
共 3 篇文章
大模型训练中,激活值常比参数更早耗尽显存。本文深入探讨选择性重计算、非重入模式、随机数状态一致性及工程门禁策略,帮助团队在保证梯度正确的前提下降低峰值显存,控制训练吞吐损失。
本文从大模型推理中容易被忽视的 CPU 到 GPU 数据通路出发,系统讲解 NUMA 亲和、页锁定内存、异步拷贝、拓扑调度与容器对齐等关键环节,并结合性能回放与上线门禁,帮助团队有效减少主机侧传输抖动、尾延迟和跨节点资源错配。
大模型推理服务常在显存尚未耗尽时因碎片化触发 OOM。本文结合 Memory Snapshot、动态 Shape、分配器调优和 vLLM 显存预算,给出可观测、可验证、可回滚的生产治理方案。