← 返回博客列表
大模型硬件

LLM 算力全景:CPU、GPU、NPU、TPU 大比拼

2026-05-15⏱ 6 分钟

LLM 算力全景:CPU、GPU、NPU、TPU 大比拼

# LLM

算力,是大模型时代的石油。理解不同计算架构的本质差异,是构建AI系统的基础。

一、为什么 LLM 需要特殊算力?

大模型的核心计算是大规模矩阵乘法(GEMM):

Attention: softmax(Q × K^T / √d) × V
FFN:       Linear(σ(Linear(x)))

这些运算的特点:

  • 高度并行:数十亿次乘加运算可同时进行
  • 规则访问模式:没有动态分支,适合流水线化
  • 内存带宽敏感:尤其在推理(Decode)阶段
  • 传统 CPU 的设计目标是低延迟、串行控制流,而大模型需要的是高吞吐、大规模并行计算——这是架构层面的根本矛盾。

    二、四大计算架构对比

    CPU(中央处理器)

    架构特征:

  • 核心数:几十个高性能核心
  • 设计哲学:低延迟、乱序执行、复杂分支预测
  • 典型代表:Intel Xeon, AMD EPYC, ARM Neoverse
  • 在 LLM 中的表现:

    场景评价
    训练大模型❌ 完全不适合(矩阵算力远不足)
    大模型推理⚠️ 仅适合轻量量化模型(llama.cpp, Ollama)
    小模型/边缘推理✅ 可接受(1-7B量化模型)
    系统调度/数据预处理✅ 必不可少

    优势: 通用性极强,内存大(服务器可达TB级),延迟低

    劣势: FP16 矩阵算力极低(远低于GPU的1/100)

    GPU(图形处理器)

    架构特征:

  • 核心数:数千个小核心(CUDA Core / Shader)+ 专用矩阵加速单元(Tensor Core)
  • 设计哲学:SIMT(单指令多线程),高吞吐并行
  • 关键内存:HBM(高带宽内存),带宽可达 3-5 TB/s
  • GPU 的 LLM 核心单元——Tensor Core:

    传统 CUDA Core:每时钟周期 1 次 FP32 乘加
    Tensor Core:每时钟周期 16×16 或 4×4 矩阵乘法(数十-数百次乘加)

    这是 GPU 在 LLM 中统治地位的根本原因。

    在 LLM 中的表现:

    场景评价
    大模型训练✅ 当前主流(H100集群是事实标准)
    大模型推理✅ 生产首选
    多模态(图像+文本)✅ 原生优势(图像计算起源)
    推理内存效率⚠️ HBM贵,容量有限(80-192GB)

    NPU(神经网络处理器)

    架构特征:

  • 专为神经网络推理设计的定制芯片
  • 内置矩阵/张量运算单元,去除通用计算冗余
  • 典型代表:Apple Neural Engine, Qualcomm Hexagon, 华为昇腾 Ascend
  • 与 GPU 的本质区别:

    GPU = 通用并行处理器(恰好很适合深度学习)
    NPU = 深度学习专用处理器(极致优化但灵活性低)

    在 LLM 中的表现:

    场景评价
    边缘/端侧推理✅ 功耗极低,集成度高(手机/笔记本)
    企业私有化部署✅ 华为昇腾在国内有竞争力
    大规模训练⚠️ 生态和工具链不如 NVIDIA
    通用 AI 开发⚠️ CUDA 生态壁垒明显

    TPU(张量处理单元)

    架构特征:

  • Google 自研 ASIC,专为 TensorFlow/JAX 设计
  • 核心是脉动阵列(Systolic Array):矩阵乘法的硬件流水线
  • 仅通过 Google Cloud 提供
  • TPU vs GPU 本质对比:

    GPU:灵活的通用并行处理器
    TPU:固定的矩阵乘法流水线
    
    TPU的矩阵乘法:更高效(专用流水线)
    TPU的灵活性:较低(算子支持有限)

    在 LLM 中的表现:

    场景评价
    Google内部大模型训练✅ Gemini等均在TPU上训练
    JAX/TensorFlow生态✅ 原生支持
    PyTorch生态⚠️ 通过XLA适配,有差距
    私有化/本地部署❌ 仅限Google Cloud

    三、主要供应商横向对比

    NVIDIA

    核心产品:H100、H200、B100/B200(Blackwell)、GH200

    指标H100 SXM5H200 SXM5B200
    Tensor Core(BF16)989 TFLOPS989 TFLOPS2,250 TFLOPS
    HBM容量80 GB141 GB192 GB
    内存带宽3.35 TB/s4.8 TB/s8.0 TB/s
    NVLink带宽900 GB/s900 GB/s1,800 GB/s
    TDP700W700W1,000W

    NVIDIA 的核心护城河:CUDA 生态

    CUDA 已经有 18年 的积累,拥有:

  • 最完整的深度学习框架支持(PyTorch、TF原生CUDA)
  • 最成熟的分布式训练工具(Megatron-LM、NCCL)
  • 最丰富的推理优化库(TensorRT、TensorRT-LLM)
  • 数百万开发者的使用习惯
  • 强: 市场占有率 ~80%,生态无可撼动,性能最强

    弱: 价格极高(H100 约 2.5-4 万美元/卡),供货紧张,功耗高

    AMD

    核心产品:MI300X、MI325X、MI350X(Instinct系列)

    指标MI300XMI325X
    FP16算力1,307 TFLOPS1,307 TFLOPS
    HBM容量192 GB256 GB
    内存带宽5.3 TB/s6.0 TB/s

    AMD 的核心差异化:超大显存

    MI300X 的 192GB HBM3 是 H100 80GB 的 2.4 倍,这使其在以下场景中有优势:

  • 超长上下文推理(无需多卡切分)
  • 大批次推理(显存装更多 KV Cache)
  • 超大模型单机推理(405B模型可在8卡MI300X上运行)
  • ROCm vs CUDA: AMD 的 ROCm 软件栈正在快速追赶,但与 CUDA 生态仍有 2-3 年差距。主流框架(PyTorch/TF)已基本支持,但工程师找bug、优化性能时依然吃力。

    强: 性价比高,显存容量领先,微软/Meta等云厂商开始大规模采购

    弱: ROCm 生态成熟度不足,优化工具链落后于 NVIDIA

    Google(TPU v5 / v6)

    代次型号算力特点
    TPU v4Pod 4096芯片1.1 EFLOPS(pod)训练Gemini Pro
    TPU v5e单芯片393 TFLOPS性价比型,推理优化
    TPU v5p单芯片459 TFLOPS训练优化
    TPU v6 (Trillium)单芯片~4x v5e2024-2025主力

    Google 的战略价值:

  • Gemini / Gemma 系列均在 TPU 上训练
  • Google Cloud 客户可获得竞争力强的 TPU 定价
  • JAX 框架在 TPU 上有极致优化(比 PyTorch+GPU 快 20-40%)
  • 强: 矩阵运算效率高,功耗比好,Google Cloud 深度整合

    弱: 仅限 GCP,PyTorch 生态适配不完美,私有化部署无解

    华为(昇腾 Ascend)

    核心产品:Ascend 910B、910C、920

    指标Ascend 910BAscend 920(预期)
    FP16算力800 TFLOPS~2,000 TFLOPS
    HBM容量64 GB96 GB
    配套软件CANN + MindSporeCANN 8.0

    战略意义大于技术本身:

    在美国出口管制(EAR)限制 H100/A100 出口中国的背景下,昇腾成为中国AI产业的战略备选。

  • 百度、阿里、华为内部均在适配昇腾集群
  • PyTorch/Transformers 通过 Torch_npu 插件支持
  • 与 NVIDIA 仍有 1-2 代 的性能差距
  • 强: 国内合规,供货有保障,政策支持力度大

    弱: 全球生态薄弱,CUDA 迁移成本高,算力密度仍落后

    Apple(Apple Silicon / Neural Engine)

    核心产品:M4 Max、M4 Ultra、M-Series Neural Engine

    芯片统一内存ANE算力GPU算力
    M4 Max128 GB38 TOPS54.7 TFLOPS
    M4 Ultra192 GB76 TOPS109 TFLOPS

    Apple Silicon 的独特价值:统一内存架构(UMA)

    传统架构:CPU内存(DDR5)+ GPU显存(HBM)互传,带宽瓶颈
    Apple UMA:CPU/GPU/ANE共享同一内存池,无传输开销

    对于 本地推理,这意味着:

  • 192GB 统一内存可运行约 100B 量化模型(Q4)
  • 内存带宽约 800 GB/s(对比H100的3.35 TB/s,但无传输损耗)
  • 功耗极低(整机 ~60-100W,H100单卡700W)
  • 典型场景: 开发者本地调试、隐私敏感推理、边缘 AI

    不适合: 大规模训练(GPU算力太低)、高并发生产推理

    四、选型决策框架

    我的需求是?
    ├── 大规模训练(>10B参数)
    │   ├── 有预算 + 不限地区 → NVIDIA H100/H200/B200 集群
    │   ├── 预算受限 + 大显存需求 → AMD MI300X
    │   └── 在Google Cloud上 → TPU v5/v6
    │
    ├── 生产推理(高并发)
    │   ├── 对CUDA生态依赖强 → NVIDIA A100/H100
    │   ├── 超长上下文/大批次 → AMD MI300X
    │   └── 成本敏感型 → NVIDIA L40S / AMD MI300E
    │
    ├── 私有化部署(企业)
    │   ├── 中国区合规优先 → 华为昇腾 910B/C
    │   └── 其他地区 → NVIDIA A100 / AMD MI250
    │
    └── 本地/边缘推理
        ├── macOS开发者 → Apple M4 Max/Ultra
        ├── 端侧AI(手机/IoT) → Qualcomm NPU / Apple ANE
        └── x86本地推理 → Intel Arc / NVIDIA RTX 4090

    五、未来趋势

    1. 内存墙(Memory Wall) 是LLM算力的核心瓶颈——不是算力不够,是带宽不足

    2. CXL 互联 将打破GPU显存孤岛,内存池化成为趋势

    3. 光子计算(Photonic Computing) 有望突破电学互联的功耗极限

    4. 国产算力 追赶速度加快,华为、寒武纪、沐曦等在2025-2026年均有重要产品发布

    总结

    NVIDIA 的 GPU 凭借 CUDA 生态的多年积累,在大模型领域具有近乎垄断的地位。但随着 AMD MI300X 的大显存优势和华为昇腾的国内需求,市场格局正在悄然变化。对于大多数团队,NVIDIA 是最优选择;对于追求成本效益或有合规要求的团队,AMD 和昇腾是值得认真评估的替代方案。

    探索智能,酷在前沿