Ch20 推理部署

AI 硬件:GPU 不是唯一的选择

一句话理解

2026 年,AI 硬件市场正在经历从 NVIDIA 一家独大到多元竞争的转变。AMD MI300X、AWS Trainium、Intel Gaudi、Cerebras、Groq LPU 等替代方案正在各自的场景中证明自己的价值。选择什么硬件,取决于你的工作负载是训练还是推理、是大规模集群还是边缘部署。

为什么需要替代硬件

NVIDIA 的优势与瓶颈

NVIDIA 在 AI 训练市场占据约 90% 的份额,其 CUDA 软件生态是二十年来积累的护城河。但 NVIDIA 面临两个问题:

  1. 供应紧张:H100/B200 的交付周期长,价格高
  2. 成本:一张 H100 约 30,000,B200 更高,大规模集群的硬件成本极高

这给了替代硬件厂商进入市场的机会窗口。

芯片类型速览

类型代表产品核心特点适用场景
GPU (通用图形处理器)NVIDIA H100/B200, AMD MI300X通用计算,CUDA 生态训练 + 推理
TPU (张量处理器)Google TPU v5p/v6大规模矩阵乘法优化训练 + 推理 (Google Cloud)
LPU (语言处理器)Groq LPU超低延迟推理实时推理
ASIC (专用芯片)AWS Trainium/Inferentia极致能效比特定场景训练/推理
Wafer-Scale (晶圆级)Cerebras CS-3超大规模单芯片大规模训练
可编程加速器Intel Gaudi 3通用加速训练 + 推理

主流替代硬件详解

AMD MI300X

AMD 最直接的 NVIDIA 挑战者。MI300X 是 CDNA3 架构的数据中心 GPU,2024 年发布。

维度AMD MI300XNVIDIA H100NVIDIA H200
显存容量192 GB HBM380 GB HBM3141 GB HBM3e
显存带宽5.3 TB/s3.35 TB/s4.8 TB/s
FP16 算力1.3 PFLOPS990 TFLOPS990 TFLOPS
互联Infinity FabricNVLink 4.0NVLink 4.0
软件生态ROCm (追赶中)CUDA (成熟)CUDA (成熟)
价格~$15,000-20,000~$25,000-30,000~$30,000-35,000

AMD 的最大优势是显存容量和带宽——192 GB 显存可以加载更大的模型,5.3 TB/s 的带宽对推理吞吐量有直接帮助。最大劣势是软件生态:ROCm 虽然快速追赶,但相比 CUDA 的成熟度和社区支持仍有差距。2026 年,ROCm 6.x 已经支持 PyTorch、vLLM、TensorRT-LLM 等主流框架,差距在缩小。

AWS Trainium / Inferentia

AWS 自研的 AI 芯片,仅供 AWS 云上使用。

芯片定位核心规格适用场景
Trainium 2训练最高 65 EFLOPS (集群), 96 GB HBM大规模分布式训练
Inferentia 2推理最高 380 TFLOPS (12 芯片), 32 GB HBM低成本推理
Inferentia 3推理 (2026)更高吞吐,更低延迟下一代推理

Trainium 的核心价值主张是成本:AWS 声称 Trainium 的训练成本比 GPU 实例低 30-40%。Inferentia 的推理成本更低。但限制也很明显——只能在 AWS 上使用,软件生态(Neuron SDK)比 CUDA 窄。

Intel Gaudi 3

Intel 通过收购 Habana Labs 进入 AI 芯片市场。Gaudi 3 是 2024 年发布的第三代产品。

维度Intel Gaudi 3NVIDIA H100
制造工艺5nm4nm (TSMC)
显存128 GB HBM2e80 GB HBM3
互联24x 200 GbE RoCENVLink + InfiniBand
软件SynapseAI (PyTorch 原生)CUDA
价格低于 H100~$25,000-30,000

Gaudi 3 的独特之处在于网络架构:集成 24 个 200 GbE 端口,不需要额外的 NIC 就能实现大规模互联。Intel 放弃了自研编程语言,直接支持 PyTorch 原生 API,降低了迁移成本。2026 年,Intel 正在推进 Falcon Shores(融合 GPU+Gaudi 架构的下一代产品)。

Cerebras CS-3

Cerebras 走了一条完全不同的路:晶圆级芯片(Wafer-Scale Engine)。一个芯片有整个晶圆那么大,集成了 4 万亿个晶体管和 90 万个计算核心。

维度Cerebras CS-3NVIDIA H100 (8-GPU 节点)
芯片面积46,225 mm² (整片晶圆)814 mm² x 8
片上内存44 GB SRAM640 GB HBM (聚合)
内存带宽21 PB/s26.8 TB/s (聚合)
互联片上 (零延迟)NVLink + NVSwitch
功耗~23 kW~5.6 kW (8x 700W)

Cerebras 的独特优势是片上内存带宽——21 PB/s,比 H100 的 HBM 高近 1000 倍。这意味着内存受限的操作(如 attention 中的 softmax)可以在极短时间内完成。但限制也很明显:44 GB 片上 SRAM 相对较小,需要特殊的模型分片策略。

Groq LPU

Groq 的 LPU (Language Processing Unit) 是专门为 LLM 推理设计的芯片,以超低延迟著称。

维度Groq LPUNVIDIA H100 (推理)
架构确定性张量流处理器SIMT GPU
内存~230 MB SRAM (per chip)80 GB HBM
每个 token 延迟极低 (<10ms)较高 (20-50ms)
吞吐量中高
编译器确定性编译器 (无调度开销)通用编译器
适合场景实时交互批量离线推理

Groq LPU 的设计哲学是用编译器替代调度器——所有计算在编译时确定,运行时没有动态调度开销。这使得延迟极低且可预测。但 ~230 MB SRAM 的片上内存是一大限制,处理大模型时需要大量芯片(例如 Llama 3 70B 需要约 500 个 LPU)。

Google TPU v5p/v6

Google 的 TPU 是最早的 AI 专用芯片之一,目前已到第六代。

维度TPU v5pTPU v6 (2025-2026)
架构脉动阵列 (Systolic Array)增强脉动阵列
互联ICI (Inter-Chip Interconnect)下一代 ICI
软件JAX, TensorFlow, PyTorch (XLA)原生多框架
可用性Google Cloud 独占Google Cloud 独占

TPU 的独特优势是与 Google 的软件栈(JAX、XLA 编译器)深度集成,在大规模训练中表现出色。Gemini 等 Google 模型都是在 TPU 上训练的。但 TPU 只能在 Google Cloud 上使用,限制了选择范围。

GPU vs TPU vs LPU vs ASIC:全面对比

维度GPU (NVIDIA)TPU (Google)LPU (Groq)ASIC (Trainium)Wafer-Scale (Cerebras)
灵活性最高高低 (仅推理)中中
训练性能最高很高N/A高最高 (特定场景)
推理延迟中低极低低极低
推理吞吐高高中高中
能效比中高很高最高高
软件生态最成熟 (CUDA)成熟 (JAX/TF)有限有限 (Neuron)有限
云可用性所有云仅 Google CloudGroq Cloud仅 AWSCerebras Cloud
购买成本高按时租用按时租用按时租用按时租用
适合场景通用Google 生态实时推理成本敏感科学计算

2026 年:NVIDIA 的统治地位正在被挑战

2026 年 AI 硬件市场的变化:

  1. AMD 在推理市场崛起:MI300X 的大显存和高带宽使其在 LLM 推理场景中表现优异,许多云厂商(Microsoft Azure、Oracle Cloud)开始提供 MI300X 实例
  2. 自研芯片成为大厂标配:Google (TPU)、Amazon (Trainium)、Microsoft (Maia)、Meta (MTIA) 都在自研 AI 芯片,减少对 NVIDIA 的依赖
  3. Groq 和 Cerebras 在特定场景中找到位置:超低延迟推理和超大规模科学计算
  4. FP4/FP8 原生支持成为新硬件的标配:B200 的 FP4、MI300X 的 FP8 原生支持,让低精度推理成为默认
  5. CUDA 护城河在变浅:PyTorch 2.x 的 torch.compile 和 OpenAI Triton 让跨硬件编程变得更容易,减少了对 CUDA 的依赖

选择指南

场景推荐硬件原因
大规模训练 (千卡级)NVIDIA H100/B200CUDA 生态最成熟,分布式训练最稳定
大规模训练 (Google 生态)TPU v5p/v6JAX 原生支持,成本可控
大规模推理 (高吞吐)AMD MI300X大显存 + 高带宽,性价比高
实时推理 (低延迟)Groq LPU确定性低延迟
成本敏感推理AWS Inferentia云上最低成本
科学计算/超大规模模型Cerebras CS-3片上内存带宽优势

相关笔记

来源

关联6

知识图谱

局部图谱 · 点节点跳转