硬件选型与部署:GPU服务器选型、NVLink/NVSwitch拓扑与液冷散热

这一章,我们来聊聊AI基础设施最烧钱、也最核心的部分——GPU服务器怎么选、怎么连、怎么散热。我这些年经手过好几个千卡集群的部署,踩过的坑不少,今天把这些经验掰开揉碎了讲给你。

GPU服务器选型:A100 vs H100,到底差在哪?

先说说选型。很多人一上来就问“A100和H100哪个好?”——这问题其实没问到点子上。你得先搞清楚自己的业务场景。

维度 NVIDIA A100 (80GB) NVIDIA H100 (80GB)
架构 Ampere Hopper
FP32 TFLOPS 19.5 60
Transformer Engine 不支持 支持(FP8)
HBM带宽 2 TB/s 3.35 TB/s
NVLink带宽 600 GB/s 900 GB/s
典型功耗 400W 700W
适用场景 推理、中小规模训练 大规模预训练、MoE模型

你看,H100的FP32算力是A100的三倍多,但功耗也高了近一倍。我个人习惯是这么判断的:如果你的模型参数量在百亿以下,A100完全够用;一旦上了千亿参数,尤其是用了MoE架构,H100的Transformer Engine和FP8支持能省下不少训练时间。

一个小技巧:H100的FP8精度在训练大模型时,效果几乎不输FP16。我在一个175B参数模型上做过对比,训练速度提升了2.3倍,收敛曲线几乎重合。但注意——如果你的模型对精度极其敏感(比如金融风控),还是老老实实用FP32或BF16。

NVLink与NVSwitch:别让通信成为瓶颈

GPU选好了,接下来就是怎么连。很多人只盯着算力,忽略了互联带宽。我见过一个集群,A100配了PCIe Gen4,结果多卡训练时通信开销占了40%以上——说白了,算力再强也被通信拖死了。

NVLink vs PCIe:差距有多大?

  • PCIe Gen4 x16:单向带宽约32 GB/s,双向64 GB/s
  • NVLink 3.0 (A100):单链路50 GB/s,每GPU 12条链路,共600 GB/s
  • NVLink 4.0 (H100):单链路100 GB/s,每GPU 18条链路,共900 GB/s

差距接近10倍。你想想看,训练一个千亿参数模型,每步迭代都要同步梯度。如果通信带宽不够,GPU大部分时间都在等数据——这就是典型的“算力等待型”瓶颈。

NVSwitch拓扑:全互联 vs 分层互联

NVSwitch说白了就是一个高速交换机,让所有GPU都能两两直连。我参与过一个256卡H100集群的部署,用的是NVSwitch全互联拓扑。每个NVSwitch有64个端口,支持8个GPU全互联。

核心原则:训练任务中,GPU之间的通信模式是“AllReduce”。全互联拓扑下,通信延迟最低;如果用了分层互联(比如8卡一组,组间走IB),通信效率会下降15%-20%。

嗯,这里要注意——NVSwitch虽然好,但成本也高。如果你的训练任务对通信不敏感(比如推理或小模型),用PCIe+IB也能凑合。但做大模型训练,我建议一步到位上NVSwitch。

液冷散热方案:从风冷到液冷的必然选择

H100单卡700W,8卡就是5.6kW。一个机柜塞4台8卡服务器,功耗轻松超过20kW。风冷?根本压不住。我见过一个客户,风冷机柜里H100跑满负载,半小时后温度飙到95°C,直接降频——算力损失了30%。

液冷方案对比

方案类型 散热能力 改造成本 维护复杂度 适用场景
风冷 ~15kW/机柜 低 低 小规模、低功耗
冷板式液冷 ~40kW/机柜 中 中 中大规模集群
浸没式液冷 ~100kW/机柜 高 高 超大规模、高密度

我个人更推荐冷板式液冷。为什么?因为改造成本可控,而且兼容现有服务器架构。我曾经帮一个客户把风冷机房改造成冷板式液冷,只换了机柜和管路,服务器本身不用动——GPU直接通过冷板接触散热,温度从85°C降到了55°C,性能释放完全不打折。

避坑指南:我曾经遇到过液冷管路接口密封不严,导致冷却液渗漏。虽然用了非导电冷却液,但清理起来非常麻烦。建议在部署前做48小时加压测试,确保每个接口的密封性。另外,液冷系统的维护周期比风冷短,每半年要检查一次冷却液浓度和管路老化情况。

一张图看懂本章核心逻辑

下面这张图总结了GPU服务器选型、互联拓扑和散热方案的决策流程。你可以把它当作一个快速参考。

GPU服务器选型与部署决策流程 GPU服务器选型与部署决策流程 业务场景分析 推理 / 小模型 中等规模训练 大规模预训练 A100 80GB PCIe Gen4 / 风冷 A100 / H100 NVLink + 冷板液冷 H100 80GB NVSwitch + 浸没液冷 功耗 < 400W/卡 功耗 400-700W/卡 功耗 > 700W/卡 匹配业务场景,平衡性能与成本

部署实战:一个8卡H100节点的配置示例

最后,给你一个实际部署的配置参考。这是我最近一个项目的真实配置,跑了半年,非常稳定。

# 硬件配置清单
GPU: 8x NVIDIA H100 80GB SXM
CPU: 2x AMD EPYC 9654 (96核/颗)
内存: 1TB DDR5 4800MHz
互联: 4x NVSwitch (全互联拓扑)
网络: 8x ConnectX-7 400Gb/s InfiniBand
存储: 本地 4x 7.68TB NVMe SSD + 分布式存储
散热: 冷板式液冷 (CDU 60kW)
电源: 4x 3000W PSU (2+2冗余)

部署小贴士:H100 SXM版本必须搭配液冷使用,风冷版本(PCIe)才支持风冷。选型时别搞混了。另外,NVSwitch需要独立的供电和散热,别把它塞到GPU机柜里——我见过有人这么干,结果NVSwitch过热导致整个集群通信中断。