模型层 开放阅读

显存利用率

Memory Utilization

概念 ID
memory-utilization
更新时间
2026-05-29
来源数量
待补

显存利用率(Memory Utilization)

3 秒看懂

显存利用率 = 实际占用显存 / 物理可用显存总量。 它衡量的是:花大价钱买来的昂贵显存(如 HBM),到底用了多少。 利用率太低 → 硬件空转浪费资本;利用率先达 100% → 训练/推理被”卡脖子”,模型上不去。

3 分钟产业解释

为什么显存利用率是 AI 产业的核心指标?

一句话:显存是 AI 芯片最贵的”场地费”。

在 AI 加速器(GPU / TPU / NPU)的 BOM(物料清单)中,HBM(高带宽显存)通常占据相当比例的芯片成本,据行业估算可占高端 AI 芯片总成本的三成至五成[行业估算,未公开精确比例]。这意味着:

产业视角影响
芯片厂商HBM 容量/带宽是产品竞争力的核心卖点
云厂商/数据中心显存利用率直接决定 GPU 实例的 ROI
模型开发者显存决定能训多大的模型、用多大的 batch
HBM 供应商(SK 海力士/三星/美光)供需关系决定议价权与产能规划

当前行业痛点:

  • 大模型训练的显存需求增速 > 显存容量增速
  • 单卡显存装不下大模型 → 被迫使用更多卡做并行 → 成本指数级上升
  • 显存利用率低(如 50%-70%)的集群,相当于每秒在烧钱

15 分钟专家深入

显存利用率的多层含义

“显存利用率”并非单一指标,它在不同语境下有不同含义:

1. 操作系统/驱动层视角

  • nvidia-smi 报告的 Memory Usage(已用显存/总显存)
  • 包含:模型参数、梯度、优化器状态、激活值、显存碎片、CUDA Context 开销
  • 这是静态占用,不反映瞬时带宽利用

2. 硬件监控层视角

  • HBM Controller 的实际读写带宽 / 峰值带宽
  • 反映的是带宽利用率(Bandwidth Utilization)
  • 与”显存容量利用率”是正交的两个维度

3. 调度/集群管理层视角

  • 一个训练任务分配到的显存 / 整个节点或集群的总显存
  • 反映资源分配效率

4. 有效利用率 vs 名义利用率

  • 名义:分配了但没有被有效计算使用(如 padding、冗余拷贝)
  • 有效:真正服务于 forward/backward 计算的显存占比

关键公式

容量利用率(Capacity Utilization):
    U_cap = Allocated_Memory / Total_Physical_Memory × 100%

带宽利用率(Bandwidth Utilization):
    U_bw = Actual_Bandwidth / Peak_Theoretical_Bandwidth × 100%

有效计算显存占比:
    U_eff = (Param + Grad + OptState + Activation) / Allocated_Memory × 100%

技术原理

显存中到底装了什么?

以典型的混合精度(bf16/fp16)LLM 训练为例,单卡显存占用构成:

┌─────────────────────────────────────────────────────┐
│                  总显存占用 (Total)                   │
├─────────────────────────────────────────────────────┤
│                                                       │
│  ┌─────────────────────────────────────────────┐     │
│  │  模型参数 (Parameters)                        │     │
│  │  - 训练时通常存 2 份: master weight (fp32)    │     │
│  │    + 计算副本 (bf16/fp16)                     │     │
│  └─────────────────────────────────────────────┘     │
│                                                       │
│  ┌─────────────────────────────────────────────┐     │
│  │  梯度 (Gradients)                             │     │
│  │  - 与参数同形状、同精度                        │     │
│  └─────────────────────────────────────────────┘     │
│                                                       │
│  ┌─────────────────────────────────────────────┐     │
│  │  优化器状态 (Optimizer States)                 │     │
│  │  - Adam: 一阶动量(m) + 二阶动量(v), fp32      │     │
│  │  - 约为参数量的 2×(fp32) = 8 bytes/param      │     │
│  └─────────────────────────────────────────────┘     │
│                                                       │
│  ┌─────────────────────────────────────────────┐     │
│  │  激活值 (Activations)                         │     │
│  │  - forward 的中间结果,用于 backward 计算       │     │
│  │  - 与 batch_size × seq_len 正相关              │     │
│  │  - 可通过 checkpointing 重计算换空间           │     │
│  └─────────────────────────────────────────────┘     │
│                                                       │
│  ┌─────────────────────────────────────────────┐     │
│  │  通信缓冲 / 临时工作区 / 碎片                   │     │
│  │  - AllReduce/ReduceScatter 的临时 buffer       │     │
│  │  - 显存碎片(fragmentation)导致不可用空间      │     │
│  └─────────────────────────────────────────────┘     │
│                                                       │
└─────────────────────────────────────────────────────┘

量化估算(以 Adam 优化器 + bf16 混合精度为例)

每参数的显存开销约为:

组成精度每参数字节数
Master Weightfp324 bytes
计算副本bf162 bytes
梯度bf162 bytes
一阶动量 (m)fp324 bytes
二阶动量 (v)fp324 bytes
合计-~16 bytes/param

粗略经验:1B 参数 ≈ 16GB 显存(仅参数+优化器,不含激活)

注意:上述为经典 AdamW 的典型估算;实际因优化器实现、是否使用混合精度策略(如 DeepSpeed ZeRO)而有显著差异。

利用率先到 100% 时发生什么?

显存利用率先达 100%
        │
        ▼
┌──────────────────┐
│ CUDA OOM Error   │  ← 最常见表现
│ "out of memory"  │
└──────────────────┘
        │
        ▼
┌──────────────────────────────────────┐
│ 系统级应对(部分可观测到)              │
│ - 频繁的显存分配/释放 → 带宽抖动        │
│ - 显存碎片 → 分配失败即使总量够          │
│ - Host-Device 数据交换增加 → 训练变慢   │
└──────────────────────────────────────┘

提升有效显存利用率的技术手段

技术机制效果
梯度检查点(Activation Checkpointing)不保存中间激活值,backward 时重计算以约 33% 计算换大幅显存节省(原文:约 [待查证])
ZeRO Stage 1/2/3分片优化器状态/梯度/参数到多卡显存占用从 O(N) 降至 O(N/k),k 为 GPU 数
混合精度训练计算用 bf16/fp16,累加用 fp32参数+梯度显存减半
梯度累积多个 micro-batch 累积后再更新减少优化器状态更新频率,间接影响显存调度
Flash Attention避免显式存储完整 Attention 矩阵激活值显存从 O(N²) 降至 O(N)
CPU Offloading将不活跃参数/状态卸载到 CPU 内存显存释放但增加 PCIe/总线延迟
量化推理INT8/INT4 权重推理时参数显存减 2-4 倍
PagedAttention (vLLM)类似虚拟内存的分页管理减少 KV Cache 碎片

技术演进史

显存容量与利用效率的代际演进

时间轴(简化)
──────────────────────────────────────────────────────────────►

2016          2018          2020          2022          2024
  │             │             │             │             │
  ▼             ▼             ▼             ▼             ▼
P100          V100          A100          H100          B200
~16GB         ~32GB         ~80GB         ~80GB         ~192GB
HBM2          HBM2          HBM2e         HBM3          HBM3e
  │             │             │             │             │
  │             │             │             │             │
  ▼             ▼             ▼             ▼             ▼

显存管理技术演进:
──────────────────────────────────────────────────────────────►
2016-2018     2019-2020     2021-2022     2023-2024
  │             │             │             │
  ▼             ▼             ▼             ▼
手动管理      混合精度      ZeRO/         Flash Attention
CUDA 碎片     Apex 开始     FSDP 出现     PagedAttention
  → 用户    → 自动混合    → 分布式显存   → 算子级显存
  自己负责     精度普及      优化成为标配    优化成热点

关键转折点

  • 混合精度训练普及(~2019):NVIDIA Apex,后 PyTorch 原生支持 → 显存效率显著提升
  • ZeRO 论文发布(2019):DeepSpeed 团队 → 分布式显存优化的理论基础
  • Flash Attention v1(2022):Tri Dao 提出 → 注意力机制的显存从 O(N²) → O(N) 的突破
  • HBM3/3e 量产(2023-2024):单颗堆叠层数增加,容量和带宽同步提升

技术路线对比

不同并行策略下的显存效率对比

策略显存分片方式通信开销显存效率适用场景
数据并行(DP)不分片,每卡全量复制AllReduce 梯度低(每卡冗余)小模型、数据量大
模型并行(张量并行,TP)按层内张量维度切分每层 AllReduce / ReduceScatter中等单节点多卡(NVLink)
流水线并行(PP)按层切分到不同设备点对点通信(pipeline bubble)中等跨节点、多层模型
ZeRO Stage 1分片优化器状态AllReduce 梯度较高显存优化入门
ZeRO Stage 2分片优化器+梯度ReduceScatter + AllGather平衡方案
ZeRO Stage 3分片一切(参数+梯度+优化器)AllGather 前向 + AllGather/ReduceScatter 反向最高(理论)超大模型
FSDP (PyTorch)类 ZeRO Stage 3类似 ZeRO-3PyTorch 生态

显存优化技术对比

技术节省的是什么计算代价实现复杂度
混合精度参数/梯度显存几乎无(可能更快)低(框架原生支持)
梯度检查点激活值显存~33% 额外计算(业界常见估算,实际因模型结构而异)低-中
CPU Offloading整体显存占用PCIe 延迟增加
Flash AttentionAttention 矩阵显存无(可能更快,减少 HBM 访问)低(库支持)
量化(INT8/4)权重显存精度损失风险中-高(需校准/量化感知训练)

上下游

产业链位置

上游                          本概念核心                        下游
─────                        ─────────                       ─────

┌─────────────┐            ┌──────────────────┐           ┌──────────────────┐
│ HBM 供应商   │            │                  │           │ 模型训练/推理     │
│ SK海力士     │───────────►│  显存利用率       │──────────►│ - LLM 训练       │
│ 三星        │  HBM 芯片   │  Memory          │  约束能力  │ - 推理服务       │
│ 美光        │            │  Utilization     │           │                  │
└─────────────┘            │                  │           └──────────────────┘
                            │  核心意义:       │
┌─────────────┐            │  决定硬件资源的   │           ┌──────────────────┐
│ 封装厂商     │            │  实际产出效率     │           │ 云厂商 / AI公司   │
│ 台积电 CoWoS │───────────►│                  │──────────►│ TCO 优化         │
│ 日月光 ASE   │  先进封装   │                  │  运营指标  │ 实例定价         │
└─────────────┘            └──────────────────┘           └──────────────────┘

┌─────────────┐                 ▲                              │
│ GPU 架构设计 │                 │                              ▼
│ NVIDIA       │─────────────────┘           ┌──────────────────┐
│ AMD          │  显存带宽/容量               │ 框架 & 编译器     │
│ 华为昇腾     │  架构决定上限               │ PyTorch/XLA      │
└─────────────┘                              │ DeepSpeed/Megatron│
                                             └──────────────────┘

关键上游约束

  • HBM 产能:当前 AI 芯片放量的主要瓶颈之一,HBM 产能被少数厂商垄断
  • 先进封装:CoWoS(台积电)等 2.5D/3D 封装产能紧俏
  • 显存带宽 vs 容量:有时不是”装不下”而是”读不完”——带宽瓶颈同样限制有效利用率

关键指标

显存利用率相关指标体系

指标定义观测工具/来源目标范围(经验性)
容量利用率已分配显存 / 物理显存nvidia-smi、DCGM80%-95%(过高易 OOM)
带宽利用率实际带宽 / 峰值带宽硬件 Profiler(Nsight, rocprof)越高越好,AI 训练常见 60%-90%
显存效率有效数据 / 总分配需 Profile 工具估算需排除碎片和冗余
Model FLOPS Utilization (MFU)实际 FLOPS / 硬件峰值 FLOPS估算(参考 PaLM 论文方法论)30%-60% 为业界较好水平
激活值占比激活显存 / 总显存框架 Profiler依赖 batch_size 和模型深度

注:MFU 虽非直接的”显存利用率”,但显存带宽往往是 MFU 的瓶颈,两者高度相关。


供需与市场数据

显存供需格局

供给侧:

  • HBM 市场高度集中:SK 海力士、三星、美光三家占据主要份额(具体市占率各机构报告口径不一,据行业估算 SK 海力士份额领先)
  • HBM 代际:HBM2 → HBM2e → HBM3 → HBM3e → HBM4(规划中)
  • 单颗堆叠层数和单层容量持续提升(具体数字以厂商公告为准)

需求侧:

  • 每代 AI 芯片显存容量持续上升(具体规格以厂商发布为准)
  • 大模型参数量级增长(数十亿 → 数千亿 → 万亿级 MoE)驱动显存需求
  • 推理侧 KV Cache 显存占用随并发用户数线性增长

供需张力:

  • HBM 产能扩张周期长(扩产 12-18 个月)
  • AI 芯片对 HBM 的需求增速快于产能增速
  • 据多家行业报告估算,HBM 供需紧张态势在 2024-2025 年持续

显存成本占比

AI 芯片中 HBM 的成本占比,因芯片代际和具体型号而异,据行业估算,高端 AI 训练芯片中 HBM 成本可占到三成至五成[行业估算,厂商未公开精确比例]。这意味着显存利用率直接关系到资本效率。


代表公司与资本映射

按产业链环节

环节代表公司与显存利用率的关系
HBM 制造SK 海力士、三星电子、美光显存硬件供应商,产能决定供给上限
GPU 架构设计NVIDIA、AMD、华为海思/昇腾显存容量/带宽架构决定利用率天花板
先进封装台积电(CoWoS)、日月光/矽品HBM 与 GPU die 的集成依赖先进封装
存储/封装配套材料三星电机、SKC、Ibiden、ShinkoHBM 封装基板、ABF 载板等
AI 框架/优化软件Microsoft(DeepSpeed)、Meta(PyTorch/FSDP)、Google(JAX/XLA)软件优化提升显存利用率
云厂商/算力平台AWS、Azure、GCP、阿里云、字节跳动显存利用率直接影响 GPU 实例 ROI

资本映射路径

投资者视角:

显存利用率低 → 需要更多 GPU 补偿 → GPU/HBM 需求增加
         │
         ├──→ NVIDIA(GPU 销售)
         ├──→ SK 海力士/三星/美光(HBM 销售)
         └──→ 台积电(封装产能)

显存利用率优化技术成熟 → 同样任务需要更少 GPU
         │
         ├──→ 利好云厂商(成本下降、利润提升)
         └──→ 短期可能压制 GPU 需求预期(长期需求仍增长)

投资逻辑

核心投资视角

视角一:显存是 AI 算力的”硬约束”

  • 不管软件怎么优化,大模型训练/推理都需要物理显存
  • HBM 供需紧张 → HBM 厂商和封装厂商受益于量价齐升

视角二:显存利用效率提升 = 解锁更大模型

  • Flash Attention、ZeRO 等技术并非减少显存需求,而是”同样的显存训更大的模型”
  • 效率提升 → 模型规模天花板上移 → 总需求继续增长
  • Jevons 悖论在 AI 显存领域的体现:效率提升 → 需求不减反增

视角三:推理侧的显存经济学

  • 推理时显存瓶颈 = KV Cache 管理
  • PagedAttention / KV Cache 量化等技术降低单用户显存占用
  • 同样的 GPU 能服务更多用户 → 推理成本下降 → 应用爆发

视角四:关注显存带宽,而非仅容量

  • 容量决定”装多少”,带宽决定”算多快”
  • HBM3/HBM3e 带宽提升是代际升级的核心卖点
  • 带宽瓶颈比容量瓶颈更难通过软件优化解决

风险提示

  • HBM 产能过剩风险:若 AI 需求增速放缓或替代技术出现
  • 新型存储技术(如 CXL 扩展内存)可能改变显存供需格局
  • 软件优化快速迭代可能部分缓解硬件瓶颈

常见误读纠偏

❌ 误读一:nvidia-smi 显示的显存占用 = 实际有效使用

纠偏: nvidia-smi 报告的是 CUDA Context 和已分配显存,包含:

  • 显存碎片(已分配但不可用)
  • CUDA 运行时自身的开销
  • 可能存在”分配但未使用”的块

实际有效利用率通常低于 nvidia-smi 显示值。精确评估需要使用 torch.cuda.memory_stats() 等工具查看 active_bytes vs reserved_bytes

❌ 误读二:显存利用率越高越好,应该打满 100%

纠偏: 显存利用率接近 100% 时:

  • 无缓冲空间:任何临时分配(如通信 buffer)都可能导致 OOM
  • 碎片风险:即使总量够,碎片化导致无法分配连续大块
  • 性能抖动:频繁的分配/释放触发显存管理开销

经验上,85%-95% 是较为健康的训练显存利用率区间。低于此区间说明资源浪费,高于此区间则 OOM 风险骤增。[经验性判断,非硬标准]

❌ 误读三:模型参数量 / 总参数量 = 显存利用率

纠偏:

  • MoE 模型的总参数量激活参数量差异巨大
  • 一次前向传播实际加载到显存的是:激活的 expert + 共享层
  • 显存利用率应关注实际驻留显存的数据,而非模型文件中的参数总数

❌ 误读四:加显存就能解决训练瓶颈

纠偏:

  • 显存带宽不足 → 即使容量够,数据读取慢 → 计算单元空等
  • 训练瓶颈可能是:计算(FLOPS)、带宽(HBM BW)、通信(NVLink/PCIe/网络)三者之一
  • 需要 Profiling 工具定位瓶颈是算力受限(compute-bound)还是带宽受限(memory-bound)

学习路径

入门级

  1. 理解 GPU 显存基本概念nvidia-smi 实操,观察训练时显存变化
  2. 阅读:PyTorch 官方文档中 torch.cuda.memory 相关 API
  3. 实验:用小模型对比不同 batch_size 下的显存占用

进阶级

  1. 阅读:DeepSpeed ZeRO 论文(ZeRO: Memory Optimizations Toward Training Trillion Parameter Models,2019)
  2. 阅读:Flash Attention 论文(FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness,2022)
  3. 实操:使用 DeepSpeed ZeRO Stage 1/2/3 对比显存占用差异
  4. 工具:学习 Nsight Systems / PyTorch Profiler 分析显存 timeline

专家级

  1. 阅读:Megatron-LM 论文中的并行策略部分(张量并行/流水线并行的显存分析)
  2. 阅读:vLLM 论文(Efficient Memory Management for Large Language Model Serving with PagedAttention,2023)
  3. 深入:理解 HBM 架构原理(堆叠结构、通道、bank、rank)与带宽计算
  4. 研究:大模型训练显存估算公式(参考各种 training calculator)

一句话总结

显存利用率是 AI 基础设施的”资本效率仪表盘”——它决定着每一块昂贵的 HBM 芯片能产出多少有效算力,是连接芯片硬件投资与模型训练产出的核心度量衡。


延伸阅读与来源

学术论文

  • Shoeybi et al., Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism, 2019
  • Rajbhandari et al., ZeRO: Memory Optimizations Toward Training Trillion Parameter Models, SC 2019
  • Dao et al., FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness, NeurIPS 2022
  • Kwon et al., Efficient Memory Management for Large Language Model Serving with PagedAttention, SOSP 2023

技术文档

  • NVIDIA CUDA Programming Guide: Memory Management 章节
  • PyTorch 文档: torch.cuda.memory_stats() API 说明
  • DeepSpeed 官方文档: ZeRO 各 Stage 说明

行业报告

  • TrendForce / Counterpoint Research 等机构的 HBM 市场报告(注意各机构数据口径差异)
  • NVIDIA/AMD/Intel 技术博客中关于内存优化的工程文章

工具

  • nvidia-smi / nvitop:实时显存监控
  • torch.cuda.memory_summary():PyTorch 显存详情
  • NVIDIA Nsight Systems / DCGM:专业级 GPU Profiling

本页所有未标注来源的具体数字均为行业通行估算或定性表述,非精确引用。技术规格以各厂商官方发布为准。

source: 公开披露与公开资料整理 本页仅用于产业链学习、信息检索和研究辅助;不构成投资建议,不预测涨跌,不提供买卖、仓位或目标价建议。
完整概念页 复盘 13 节结构 公司投研页 沿产业链找到受益公司 投资课 把概念转成可跟踪模型