TPU v7 Ironwood vs NVIDIA B200

Google 第七代 TPU 与 NVIDIA Blackwell 全方位参数对比 · 训练 & 推理

Google TPU v7 · Ironwood · 2025 NVIDIA B200 · Blackwell · 2024

一句话看懂

TPU v7 是什么

Google 自研的第 7 代 AI 加速器,代号 Ironwood。专为大模型推理设计(Google 官方叫它 "the first TPU for the age of inference"),但也支持训练。采用双 chiplet 设计,3nm 工艺,192 GB HBM3e。通过 ICI(Inter-Chip Interconnect)互连,最大可组成 9,216 芯片的 Pod,共享 1.77 PB 内存。

B200 是什么

NVIDIA Blackwell 架构的旗舰 GPU,2024 年发布。同样采用双 die 设计,4nm 工艺,208B 晶体管,180-192 GB HBM3e。通过 NVLink 5 互连(1.8 TB/s),标准配置为 DGX B200(8 GPU)或 GB200 NVL72(72 GPU 液冷机架)。拥有完整的 CUDA 生态。

对比口径说明:以下所有数据均为单芯片 (per chip/GPU) 级别对比,数据来自官方文档和技术白皮书。B200 算力数据使用 Dense(密集)模式,不含 Sparsity 加成。

核心规格对比

规格项 TPU v7 (Ironwood) NVIDIA B200 对比
架构
代号 / 架构 Ironwood Blackwell
制程工艺 TSMC N3P (3nm) TSMC 4NP (4nm) TPU 领先
芯片设计 双 Chiplet每 chiplet 96 GB HBM 双 Die10 TB/s NV-HBI 互联 相似
核心单元 2 TensorCore + 4 SparseCoreMXU 256×256 148 SM / 528 Tensor Cores16,896 CUDA Cores
计算性能 (Dense, per chip)
BF16 (Tensor) 2,307 TFLOPS 2,250 TFLOPS TPU +3%
FP8 (Tensor) 4,614 TFLOPS 4,500 TFLOPS TPU +3%
INT8 (Tensor) 未公开具体数据 4,500 TOPS
FP4 (Tensor) 不支持 9,000 TFLOPS B200 独有
FP32 TPU 无通用 FP32 75 TFLOPS
FP64 TPU 无 FP64 37 TFLOPS B200 独有
内存
HBM 容量 192 GB HBM3e8-Hi stacks 180 GB HBM3e1000W 版本 TPU +7%
HBM 带宽 7.4 TB/s 8.0 TB/s B200 +8%
芯片间互连
互连技术 ICI (Inter-Chip Interconnect)3D Torus Mesh NVLink 5GPU-to-NVSwitch
单芯片互连带宽 1.2 TB/s 1.8 TB/s B200 +50%
外部网络
单芯片外部网络 100 GbpsDCN 400 GbpsConnectX-7 B200 4×
功耗
TDP (单芯片) ~600W官方参考值 1,000W TPU 低 40%
能效比 (FP8) ~7.7 TFLOPS/W ~4.5 TFLOPS/W TPU +71%
系统规模
标准系统 256-chip Pod最小部署单元 DGX B200 (8 GPU)10U 机箱
最大配置 9,216 chips144 racks, 1.77 PB 共享内存 NVL72 (72 GPU)液冷机架 TPU 规模更大

关键指标可视化

BF16 算力 (TFLOPS, Dense)

TPU v7 2,307 B200 2,250

FP8 算力 (TFLOPS, Dense)

TPU v7 4,614 B200 4,500

HBM 容量 (GB)

TPU v7 192 GB B200 180 GB

HBM 带宽 (TB/s)

TPU v7 7.4 TB/s B200 8.0 TB/s

芯片间互连带宽 (TB/s)

TPU v7 1.2 TB/s (ICI) B200 1.8 TB/s (NVLink 5)

能效比 FP8 (TFLOPS/W)

TPU v7 ~7.7 TFLOPS/W B200 ~4.5 TFLOPS/W

训练场景分析

大模型预训练 (100B+ 参数)

大规模预训练是两者都在争夺的核心战场。关键看三个维度:算力密度互连带宽(决定多芯片扩展效率)、内存容量(决定单芯片能装多大模型)。

  • 算力对等:BF16 和 FP8 两者基本打平(TPU v7 微领先 ~3%),单芯片层面没有显著差距。
  • 扩展性差异巨大:TPU v7 的 Pod 最大支持 9,216 chips(1.77 PB 共享内存),通过 ICI 3D Torus 直连,对超大模型训练(如 Gemini 级别)有天然优势。B200 的 NVL72 最大 72 GPU,跨节点需走 InfiniBand/Ethernet,通信开销更大。
  • 软件栈:TPU 训练使用 JAX/XLA,自动并行化能力强;B200 训练生态更广(PyTorch + Megatron-LM / DeepSpeed / FSDP),社区资源丰富。

大模型微调 (Fine-tuning)

微调通常不需要超大规模集群,更看重单节点性能和易用性。

  • TPU v7:最小部署 256 chips,对小规模微调来说资源粒度偏大。但 Google Cloud 提供按需付费的 TPU VM,实际使用中可以灵活选择 4-chip 配置。
  • B200:DGX B200 (8 GPU) 或单卡即可开始微调,FP4 精度支持让推理精度的模型可以直接微调,灵活性高。CUDA 生态下的 LoRA / QLoRA 工具链成熟。
TPU v7 训练优势:超大规模训练(数千芯片级别)的扩展性、能效比(7.7 vs 4.5 TFLOPS/W)、Google Cloud 深度集成(GKE + Cloud TPU API)。
B200 训练优势:CUDA 生态成熟度、FP4 精度支持、灵活的部署规模(单卡到 NVL72)、社区工具链丰富(Hugging Face / DeepSpeed / vLLM)。

推理场景分析

LLM 在线推理 (Serving)

推理场景最核心的指标是 tokens/s/W(每瓦吞吐)和 首 token 延迟。Google 官方称 TPU v7 为 "the first TPU for the age of inference",重点优化了推理场景。

  • 能效比:TPU v7 的 FP8 能效比达 7.7 TFLOPS/W,比 B200 高 71%。在大规模推理服务中,电费是主要成本,这一差距会直接反映在 TCO 上。
  • 内存容量:192 GB vs 180 GB,TPU v7 单芯片可以装更大的模型。对 70B 级模型的推理部署,TPU v7 需要的芯片数量更少。
  • 量化支持:B200 支持 FP4 推理(9 PFLOPS),对 INT4/FP4 量化模型的推理吞吐是 FP8 的 2 倍。TPU v7 不支持 FP4,这是一个劣势。
  • Pod 级推理:超大模型(如 Gemini Ultra / GPT-4 级别)需要跨多芯片部署。TPU v7 的 9,216-chip Pod 和 ICI 直连在这类场景下优势明显。

批量推理 / 离线推理

批量推理更看重吞吐量和成本效率,对延迟不敏感。

  • TPU v7:高能效 + 大规模 Pod 组网,适合海量请求的批量处理。Google 内部的 Gemini API 推理就运行在 TPU 上。
  • B200:FP4 推理吞吐翻倍,TensorRT-LLM 的 batching 优化成熟。对第三方模型(Llama, Mistral 等)的兼容性更好。
TPU v7 推理优势:能效比领先 71%、192 GB 大内存、Pod 级别的超大规模推理能力、Google Cloud 一体化部署。
B200 推理优势:FP4 支持(推理吞吐翻倍)、TensorRT-LLM 优化、更广泛的模型兼容性、灵活的单卡/多卡部署。

生态与软件栈

TPU v7 软件生态

  • 框架:JAX / TensorFlow(原生)、PyTorch/XLA
  • 并行化:XLA 自动分片 (GSPMD)、FSDP
  • 推理框架:JetStream、vLLM (TPU 后端)
  • 编排:GKE + Cloud TPU API、Multislice
  • 优势:XLA 编译器全自动优化,无需手动调 kernel
  • 劣势:社区生态相对封闭,第三方库支持有限

B200 软件生态

  • 框架:PyTorch、TensorFlow、JAX(均原生 CUDA)
  • 并行化:Megatron-LM、DeepSpeed、PyTorch FSDP
  • 推理框架:TensorRT-LLM、vLLM、SGLang
  • 管理:NVIDIA AI Enterprise、Mission Control
  • 优势:CUDA 生态无可替代,几乎所有 AI 论文先发 GPU 版
  • 劣势:手动优化工作量大,kernel tuning 门槛高

系统级对比

TPU v7 Pod (9,216 chips)

  • 算力 FP8:42.5 EFLOPS
  • 算力 BF16:21.3 EFLOPS
  • 总内存:1.77 PB(全局可寻址)
  • 机架数:144 racks
  • 互连拓扑:3D Torus + OCS 光交换
  • 总功耗:~10 MW
  • 每 rack:64 chips / 16 trays

DGX B200 (8 GPU)

  • 算力 FP8:36 PFLOPS (Dense)
  • 算力 BF16:18 PFLOPS (Dense)
  • GPU 总内存:1,440 GB
  • CPU:2× Xeon Platinum 8570
  • NVLink 聚合:14.4 TB/s
  • 网络:8× ConnectX-7 400G
  • 系统功耗:~14.3 kW
规模差异说明:TPU v7 Pod (9,216 chips) 和 DGX B200 (8 GPU) 不是同级别的对比。若等效比较,9,216 ÷ 8 = 1,152 台 DGX B200 才能匹配 TPU Pod 的芯片数。但两者的互连架构完全不同——TPU Pod 内所有芯片通过 ICI 直连,而 1,152 台 DGX 之间只能通过 InfiniBand/Ethernet 连接,通信效率有本质差距。

维度总结

BF16 / FP8 算力
基本持平
TPU v7 微领先 ~3%,无显著差距
FP4 / 低精度推理
B200
9 PFLOPS FP4,TPU v7 不支持
HBM 容量
TPU v7
192 GB vs 180 GB (+7%)
HBM 带宽
B200
8.0 TB/s vs 7.4 TB/s (+8%)
芯片间互连
B200
NVLink 5 1.8 TB/s vs ICI 1.2 TB/s
系统级扩展性
TPU v7
9,216-chip Pod,ICI 全直连
能效比
TPU v7
7.7 vs 4.5 TFLOPS/W (+71%)
软件生态
B200
CUDA 生态不可替代
部署灵活性
B200
单卡到 NVL72,多云部署
超大规模训练
TPU v7
Pod 直连 + 1.77 PB 共享内存
大规模推理
TPU v7
高能效 + Pod 级推理
通用 AI 研发
B200
生态广、上手快、模型兼容性强

数据来源