1. 课程导入

1.1 学习目标与课程定位

本课程面向需要真正动手搭建AI基础设施的运维工程师、平台工程师与算法工程师,聚焦于训练集群与推理集群从硬件选型到软件栈落地的完整过程。完成本章学习后,你将能够:

  • 清晰区分训练集群与推理集群在硬件配置、网络拓扑、软件栈上的核心差异;
  • 理解集群部署中涉及的关键组件(GPU/NPU、RDMA网卡、高性能存储、调度器)及其协作关系;
  • 掌握从单机多卡到多机多卡扩展时的网络与存储瓶颈判断方法;
  • 建立对后续章节(驱动安装、容器化、分布式框架配置、性能调优)的整体认知框架。
学习建议:本章不要求你记住所有硬件参数,但务必理解集群的分层架构——计算层、网络层、存储层、软件层之间的依赖关系。后续所有排障与调优,本质上都是在这四层之间定位问题。

1.2 课程结构总览

本课程共30章,按“基础→搭建→优化→实战”四个阶段递进。前5章夯实硬件与网络基础,第6-15章完成训练集群的软件栈部署,第16-22章转向推理集群的轻量化与高并发设计,第23-28章覆盖监控、容错与安全,最后两章通过综合案例串联全部知识点。

阶段 章节范围 核心交付物
基础篇 第1-5章 硬件选型清单、网络拓扑设计、BIOS与固件配置
训练集群搭建 第6-15章 可运行的分布式训练环境(含Slurm/K8s调度)
推理集群搭建 第16-22章 支持高并发低延迟的推理服务(含TensorRT/ vLLM)
运维与实战 第23-30章 监控告警体系、故障演练报告、端到端部署案例

1.3 AI训练推理集群概述

AI集群的本质是将大量加速卡(GPU/NPU)通过高速网络互联,并配合分布式存储与调度系统,对外提供统一算力。训练集群与推理集群虽然共享底层硬件,但设计哲学截然不同:

  • 训练集群追求高带宽、低延迟的集合通信(如All-Reduce),因为每轮迭代都需要同步梯度,网络性能直接决定GPU利用率;
  • 推理集群更关注吞吐量与响应延迟的平衡,常采用模型并行、动态批处理、量化等手段,对网络带宽要求低于训练,但对调度灵活性要求更高。

一个典型的AI集群硬件栈包含四个层次:

  1. 计算节点:每台服务器插有8张GPU(如NVIDIA H800/A800或华为昇腾910B),通过PCIe Switch与CPU互联;
  2. 网络层:采用InfiniBand或RoCE(RDMA over Converged Ethernet)组网,带宽通常为400Gbps/端口,拓扑多为Fat-Tree或Torus;
  3. 存储层:并行文件系统(如Lustre、BeeGFS)或分布式对象存储(如MinIO),提供数十GB/s的聚合读写带宽;
  4. 软件层:包含NVIDIA驱动、CUDA/ROCm、容器运行时(Docker+Enroot或Singularity)、作业调度器(Slurm/K8s)、分布式框架(PyTorch DDP/DeepSpeed)。
AI训练推理集群分层架构与数据流 软件层 PyTorch DDP / DeepSpeed Slurm / K8s CUDA / ROCm Docker / Enroot 驱动 + 固件 (GPU/NPU) 计算节点层 (每台服务器) 节点 A (8×GPU) CPU + PCIe Switch H800 / 昇腾910B 节点 B (8×GPU) CPU + PCIe Switch H800 / 昇腾910B 节点 C (8×GPU) CPU + PCIe Switch H800 / 昇腾910B 网络层 Spine 交换机 (400G) Leaf 交换机 (400G) RDMA / RoCE 网卡 InfiniBand / RoCEv2 · Fat-Tree 拓扑 存储层 并行文件系统 Lustre / BeeGFS 对象存储 MinIO / Ceph RGW NVMe 缓存层 聚合带宽 ≥ 50GB/s 下发作业 梯度同步 / 张量并行 检查点 / 数据集读取 软件层 计算节点 网络层 存储层

1.4 典型应用场景与部署案例

不同业务场景对集群的配置要求差异巨大,以下是三个最常见的部署形态:

场景一:大语言模型(LLM)训练集群

  • 规模:512~2048张GPU,采用3D并行(数据并行+张量并行+流水线并行);
  • 网络要求:必须使用InfiniBand,且每张GPU至少配1个400G端口,否则通信将严重拖慢训练;
  • 存储:需要高性能并行文件系统存放检查点(Checkpoint),单次保存可达数百GB;
  • 典型软件:Slurm + PyTorch + Megatron-LM/DeepSpeed。

场景二:CV/多模态推理集群

  • 规模:8~32张GPU,通常配合K8s进行弹性伸缩;
  • 网络要求:RoCE即可满足,延迟敏感度低于训练;
  • 关键指标:P99延迟 < 100ms,吞吐量 > 1000 QPS;
  • 典型软件:K8s + Triton Inference Server + TensorRT。

场景三:科研机构混合负载集群

  • 规模:64~128张GPU,同时运行训练与推理任务;
  • 核心挑战:资源隔离与优先级调度,避免推理任务被训练任务饿死;
  • 解决方案:Slurm的QOS(Quality of Service)配合cgroup隔离,或K8s的Namespace+ResourceQuota。

1.5 常见误区与设计复盘

在规划集群时,初学者常犯以下错误,需要特别警惕:

误区一:盲目追求GPU型号而忽视网络瓶颈

例如购买了8张H800,却只配置了25Gbps以太网。在数据并行训练中,每步迭代的梯度同步时间将超过计算时间,导致GPU利用率不足30%。正确做法:先估算通信量(梯度大小×参数数量),再反推所需网络带宽。

误区二:存储系统未做读写分离

将训练数据、检查点、日志全部放在同一存储池。当训练任务周期性写入检查点时,数据读取延迟飙升,导致数据加载成为瓶颈。正确做法:热数据(训练集)放在NVMe缓存,检查点写入独立的并行文件系统,日志走低性能存储。

误区三:忽略CPU与内存的配置

GPU服务器并非只堆GPU即可。数据预处理(如JPEG解码、Tokenization)非常消耗CPU。若CPU核心数不足或内存带宽不够,GPU仍会因等待数据而空闲。经验值:每张GPU至少配16个物理核心和256GB内存。