Skip to content

硬件基础速查

本页速览 推理加速硬件体系结构:GPU/SM/Tensor Core、NVIDIA 三代数据中心卡 A100/H100/H200/B200 显存带宽算力功耗对比、消费卡对比、AMD MI300X、Apple Silicon、昇腾 910B、Intel AMX、NPU,及训练/推理选型逻辑。

硬件基础速查

推理加速的所有软件优化,最终都要落到具体硬件上:SM 数、Tensor Core 算力、HBM 带宽、NVLink 互联——这些数字决定了"理论上限"。但硬件远不止参数表,它还有微架构特性(Hopper 的 TMA、Blackwell 的 FP4 Tensor Core)、互联拓扑(NVSwitch vs PCIe)、功耗与散热约束。本页是站内GPU 体系结构与优化显存层次与带宽墙Roofline 模型分布式推理等章节的"硬件参考底座",与基准数据与工具档案互为印证。

使用建议

本文不是选型采购指南,而是"硬件参数 + 微架构直觉"的速查。先读第二节建立"GPU 是什么"的心智模型,再看第三节的三代对比表,最后按需查其他芯片。参数会随新品更新而变化,正式选型请以厂商当前官方规格为准,本页数字截至 2026-08。

一、GPU 体系结构基础

GPU 能做大规模并行计算,靠的是把"算术"塞进硬件级并行执行单元。理解推理优化,必须先建立 GPU 微架构的心智模型。详见GPU 体系结构与优化

1.1 SM / warp / SIMT

SM(streaming multiprocessor) 是 GPU 的计算基本单元。一颗 GPU 由若干 SM 组成,每 SM 内部含:若干 CUDA core(标量 FP32/FP64 单元)、若干 Tensor Core(矩阵乘加专用单元)、共享内存/L1(片上高速 SRAM)、寄存器堆、以及线程调度器。

GPU 执行模型叫 SIMT(single instruction, multiple threads):单条指令派发给多个线程同时执行,类似 SIMD 但每个线程有自己的寄存器和程序计数器。每 32 个线程组成一个 warp,warp 是 SM 的最小调度单位——同一 warp 内的线程必须在同一条指令上同步执行,若分支不一致(warp divergence),未走分支的线程只能空转。

1.2 Tensor Core 与矩阵乘

Tensor Core 是 Volta 架构(V100,2017)起引入的矩阵乘加专用单元。一个时钟周期内,单个 Tensor Core 完成一次 D = A×B + C 的 4×4×4 矩阵乘加(FP16 输入、FP16/FP32 累加)。A100 的 Tensor Core 支持 BF16、FP16、TF32、INT8、INT4;Hopper 起加入 FP8(E4M3/E5M2) 原生;Blackwell 又加入 FP4 与 FP6。

LLM 推理的几乎全部算力开销在 GEMM(general matrix multiply)上:prefill 是 Q/K/V = X × W 大 batch GEMM(compute-bound),decode 是 x × W 小 batch GEMM(memory-bound,详见显存层次与带宽墙)。Tensor Core 算力直接决定 prefill 上限。

1.3 存储层次与带宽

GPU 有三级存储,延迟与带宽逐级下降:

层级容量(A100 单卡)带宽量级用途
寄存器256 KB / SM × 108 = 27 MB~30 TB/s单线程私有变量,最快
共享内存 / L1(SRAM)192 KB / SM~20 TB/swarp/block 内共享,kernel 内 tile
L2 cache40 MB(全 SM 共享)~7 TB/s跨 SM 一致性、HBM 缓冲
HBM(外显存)80 GB2.0 TB/s模型权重、KV cache、激活

LLM 推理 decode 阶段的瓶颈就在"从 HBM 读权重到 Tensor Core"这一步——每生成一个 token,要把整个模型权重过一遍。详见Roofline 模型。所以 HBM 带宽是 LLM 推理最重要的硬件指标之一。

1.4 cuDNN / cuBLAS / NCCL

NVIDIA 提供一整套底层库:

  • cuBLAS:BLAS 矩阵乘库,GEMM 的基础实现,TensorRT/PyTorch 的 GEMM 底层都走它;
  • cuDNN:深度学习算子库,卷积、池化、归一化、激活函数等的高性能实现,含 kernel auto-tuning;
  • NCCL:多卡集合通信库(AllReduce、AllGather、Broadcast),PyTorch DDP/FSDP 与 TP/PP 的底层。

这些库是"用好 GPU"的标配——自己写 CUDA kernel 通常打不过它们,优化方向应是"调它/调用它的更高层封装",而非重写。详见TensorRT 与 GPU 推理

二、NVIDIA 数据中心 GPU 三代对比

NVIDIA 是数据中心 AI 算力的事实主流,三代产品覆盖 2020–2026 的几乎所有部署场景。详见各代基准数据

2.1 微架构与代表型号

微架构年份代表数据中心卡关键新增
Ampere2020A100 40GB / 80GBTF32、稀疏 2:4、MIG
Hopper2022H100 SXM / PCIe、H200FP8 原生、Transformer Engine、TMA、多实例 GPU
Blackwell2024B200、GB200 NVL72FP4 原生、第二代 Transformer Engine、NVLink Switch 1.8 TB/s

2.2 关键参数大表

下表是三代的横向对比,数字均为 SXM 版本(PCIe 版带宽与功耗略低),以官方发布规格为准。

参数A100 80GBH100 SXM 80GBH200 141GBB200
微架构AmpereHopperHopperBlackwell
SM 数108132132148
HBM 类型HBM2eHBM3HBM3eHBM3e
HBM 容量80 GB80 GB141 GB192 GB
HBM 带宽2.0 TB/s3.35 TB/s4.8 TB/s8.0 TB/s
L2 cache40 MB50 MB50 MB60 MB
FP16/BF16 稠密算力312 TFLOPS989 TFLOPS989 TFLOPS2250 TFLOPS
FP16/BF16 稀疏算力624 TFLOPS1979 TFLOPS1979 TFLOPS4500 TFLOPS
FP8 算力1979 TFLOPS(稀疏)1979 TFLOPS(稀疏)9000 TFLOPS(稀疏)
FP4 算力18000 TFLOPS(稀疏)
INT8 算力(稀疏)1248 TOPS3958 TOPS3958 TOPS~9000 TOPS
NVLink 带宽600 GB/s900 GB/s900 GB/s1800 GB/s
PCIeGen4 ×16Gen5 ×16Gen5 ×16Gen5 ×16
TDP400 W700 W700 W1000 W
制程7 nm4 nm(TSMC N4)4 nm4 nm(TSMC N4P)

算力单位 TFLOPS/TOPS 与稀疏

TFLOPS(每秒万亿次浮点运算)针对浮点;TOPS(每秒万亿次运算)针对整数。表格里的"稀疏算力"是 NVIDIA 2:4 结构化稀疏启用后的理论峰值——每 4 个元素中 2 个为零可加速 2×,需模型权重满足稀疏度才能吃到。LLM 推理一般直接用稠密算力评估。详见剪枝与稀疏化

2.3 三代关键差异

Ampere A100(2020,详见基准数据):FP16 训练主力、TF32 是巧妙的"FP19 兼容 FP32 格式"使 FP32 训练免改代码即可加速。HBM2e 2.0 TB/s 带宽在 70B LLM decode 时已是瓶颈。MIG 把卡切成 7 个隔离实例,多租户场景好用。

Hopper H100/H200(2022/2024):硬件 Transformer Engine 自动选 FP8/FP16/BF16 精度,每层独立;TMA(tensor memory accelerator) 把"从 HBM 异步搬运 tile 到共享内存"硬件化,让 kernel 写起来更简单且更快;FP8 原生是 LLM 量化的分水岭——精度接近 FP16、算力翻倍。H200 把 HBM 升到 HBM3e 4.8 TB/s 与 141 GB,专门解决 LLM decode 带宽墙与长上下文容量。详见权重量化与混合精度

Blackwell B200(2024,详见基准数据):FP4 原生 Tensor Core,理论算力比 H100 FP8 又翻倍;HBM3e 192 GB 单卡能装下完整 405B 模型;NVLink Switch 把 72 张 B200 互联成"一台超大 GPU",TB 级参数模型单机推理可行。GB200 NVL72 整机 FP8 算力 720 PFLOPS。

三、消费卡 vs 数据中心卡

消费卡不是"阉割版数据中心卡",而是另一种设计取向:显存小但单精度算力不低、互联弱(无 NVLink)、价格低一个数量级。对个人学习与小规模推理有价值。详见llama.cpp 与 GGUF

参数RTX 4090RTX 5090A100 80GBH100 SXM
微架构Ada LovelaceBlackwell (GB202)AmpereHopper
显存24 GB GDDR6X32 GB GDDR780 GB HBM2e80 GB HBM3
显存带宽1008 GB/s1792 GB/s2.0 TB/s3.35 TB/s
FP16 稠密算力165 TFLOPS(不含稀疏)~210 TFLOPS312 TFLOPS989 TFLOPS
FP8 算力1979 TFLOPS(稀疏)
NVLink✗(仅 PCIe)✓ 600 GB/s✓ 900 GB/s
TDP450 W575 W400 W700 W
建议用途7B 以下 LLM 推理13B–30B LLM 推理训练 / 70B LLM 推理大规模训练与推理

消费卡的两个坑

显存是天花板:RTX 4090 24 GB 跑 FP16 的 Llama-3-8B(16 GB)刚好,70B 必须 INT4 量化 + offload 才勉强。互联弱:消费卡无 NVLink,多卡只能走 PCIe,TP 推理带宽瓶颈严重,2×4090 跑 70B 不如 1×H100。详见分布式推理(TP/PP)

四、AMD Instinct MI300X

AMD 的数据中心 AI 加速器,对标 H100。MI300X(2023)采用 CDNA 3 架构,HBM3 192 GB / 5.3 TB/s 带宽——单卡显存比 H100 80GB 多 2.4 倍,能直接装下完整 Llama-3-70B 甚至 405B 量化版。FP16 稠密算力 1307 TFLOPS,FP8 算力 2615 TFLOPS,均略高于 H100。

软件栈是 ROCm(对标 CUDA),PyTorch 与 vLLM 已原生支持。短板是生态成熟度与社区资料不如 CUDA,调试与算子优化工具链仍在追赶。详见推理引擎选型对比

五、Apple Silicon

Apple M2/M3 Ultra 的 Neural Engine(NPU) 主打端侧推理,统一内存架构(unified memory)让"GPU 直接访问主存"省去了拷贝,这对 LLM 推理特别友好——M3 Ultra 192 GB 统一内存可跑 Llama-3-70B 量化版,llama.cpp 对 Metal 后端原生支持良好。详见llama.cpp 与 GGUF移动端部署

型号统一内存内存带宽Neural EngineMetal 加速
M2 Ultra192 GB800 GB/s32 TOPS(INT8)
M3 Ultra192 GB800 GB/s36 TOPS(INT8)

Mac 跑 LLM 的定位

Mac 适合个人本地推理端侧原型:统一内存让"大显存"几乎免费(M3 Ultra 192GB ≈ 一张 H100 价但显存翻 2.4 倍),但算力远低于数据中心卡,吞吐量级差距大。批量服务仍要 GPU。详见移动端部署

六、华为昇腾 910B

国产 AI 加速器代表,达芬奇架构,CANN 软件栈(对标 CUDA)。910B 单卡 FP16 算力约 320 TFLOPS、HBM 64 GB / 1.6 TB/s,对标 A100 但显存略小。配套的 MindIE 推理引擎与 MindSpore 训练框架是国内合规场景的替代方案,PyTorch 通过 torch_npu 适配。短板是生态与开源工具链成熟度,社区资料与 CUDA 比仍有差距。详见推理引擎选型对比

七、Intel CPU 与 AMX

Intel Sapphire Rapids(2023)起的 AMX(Advanced Matrix Extensions) 指令集给 CPU 加了矩阵乘专用单元——本质是把 Tile 寄存器 + TDP(tile matrix multiply)指令塞进 x86。INT8/BF16 矩阵乘吞吐量相比 AVX-512 成倍提升。AMX 让 CPU 不再是"勉强能跑",而是小模型推理的现实选项:BERT-Base INT8 在 Sapphire Rapids 60 核上能做到 ~1000 tokens/s。详见OpenVINO 与 CPU 推理

CPU 推理的优势不在吞吐而在 易部署——任何服务器都有 CPU,免装 GPU、免 PCIe、免驱动。低延迟小批量场景或分布式系统的边节点推理常见。配合 OpenVINO 与 ONNX Runtime 的图优化能进一步压性能。

八、NPU 与边缘 AI 芯片

NPU(neural processing unit)泛指为神经网络推理设计的专用 ASIC,广义含 TPU、昇腾、Apple Neural Engine、高通 Hexagon 等。边缘场景的代表:

芯片厂商算力用途
Hexagon NPUQualcomm骁龙 8 Gen3 起 45 TOPS手机端 LLM、Stable Diffusion
APU(APU 融合)MediaTek天玑 9300 起 INT8 算力手机端生成式 AI
Edge TPUGoogle4 TOPS(INT8)Coral 设备、轻量分类检测
Jetson OrinNVIDIA275 TOPS(稀疏 INT8)机器人、自动驾驶边缘
Rockchip RK3588瑞芯微6 TOPS(NPU)边缘网关、轻量视觉

边缘推理的核心约束是 功耗(手机 5W、IoT 1W 量级)与 内存(手机 8–16 GB 与系统共享)。LLM 端侧落地依赖 1–4B 参数小模型 + INT4 量化 + 推理引擎极致优化。详见移动端部署

九、关键参数速查表

把上述硬件的关键参数汇总成一张表,方便横向比较。算力与带宽均以官方发布规格为准。

硬件类型显存/内存带宽FP16 算力FP8 算力TDP
A100 80GB SXMGPU80 GB HBM2e2.0 TB/s312 TFLOPS400 W
H100 SXM 80GBGPU80 GB HBM33.35 TB/s989 TFLOPS1979 TFLOPS(稀疏)700 W
H200 141GBGPU141 GB HBM3e4.8 TB/s989 TFLOPS1979 TFLOPS(稀疏)700 W
B200GPU192 GB HBM3e8.0 TB/s2250 TFLOPS9000 TFLOPS(稀疏)1000 W
RTX 4090GPU24 GB GDDR6X1008 GB/s165 TFLOPS450 W
RTX 5090GPU32 GB GDDR71792 GB/s~210 TFLOPS575 W
MI300XGPU192 GB HBM35.3 TB/s1307 TFLOPS2615 TFLOPS(稀疏)750 W
M3 UltraSoC+NPU192 GB 统一800 GB/s27 TFLOPS(GPU)~370 W(整机)
昇腾 910BNPU64 GB HBM1.6 TB/s~320 TFLOPS310 W
Sapphire Rapids XeonCPUDDR5 8 通道~460 GB/sAMX BF16 ~50 TFLOPS350 W
Snapdragon 8 Gen3SoC+NPULPDDR5X 共享~77 GB/s~5 W(NPU)

十、选型逻辑

不同场景硬件选型关注点完全不同,先分清是 训练 还是 推理、是 数据中心 还是 工作站 还是 端侧

10.1 训练 vs 推理

维度训练推理
关键瓶颈算力 + 互联(TP/PP 多卡)带宽 + 显存容量(LLM decode)
显存需求激活 + 优化器状态,要大模型权重 + KV cache
精度FP16/BF16 为主,少量 FP32INT8/INT4/FP8 为主,FP16 备选
互联要求TP 必须 NVLinkTP 必须 NVLink,单卡可无
选卡取向H100/H200/B200 高算力强互联看 batch:大 batch 重算力,小 batch 重带宽

LLM 推理的特殊性在于 decode 阶段访存密集——单卡吞吐基本等于 带宽 / 模型字节数。所以 HBM 带宽是 LLM 推理选卡最重要指标之一,甚至比算力重要。详见Roofline 模型显存层次与带宽墙

10.2 数据中心 vs 工作站 vs 端侧

数据中心:A100/H100/H200/B200 + NVLink + NVSwitch。吞吐与多卡互联是命脉,单机 8 卡 NVSwitch 拓扑是 70B+ 模型 TP 推理的标配;B200 NVL72 把 72 卡连成"一台超大 GPU",TB 级参数模型单机推理可行。功耗与散热(液冷)是部署难点。详见分布式推理(TP/PP)

工作站:RTX 4090/5090 或 Apple M Ultra。显存小但便宜、无 NVLink 但 PCIe 够单卡用。适合个人学习、原型验证、小批量服务。详见llama.cpp 与 GGUF

端侧:手机 NPU、Jetson、Edge TPU。算力小、内存小、功耗极敏感。必须 INT4 量化 + 小模型,配合 llama.cppONNX Runtime Mobile。详见移动端部署

选型三个经验法则

  1. 先定 batch 规模再选卡:单用户聊天 batch≈1 重带宽,离线批处理 batch 大重算力;
  2. 显存容量是硬墙:模型 + KV cache 装得下才有后续,装不下必须量化或 TP;
  3. 互联决定多卡上限:TP 推理 NVLink 是硬指标,PCIe 多卡只适合独立服务不通信。

延伸阅读