集群 GPU 概览

集群化 GPU 针对人工智能 (AI) 和机器学习 (ML) 工作负载进行了优化,这些工作负载优先考虑大规模分布式训练、多主机推理和复杂的高性能计算 (HPC) 任务。

借助集群化 GPU,团队可以使用专用网络结构和同步维护,将数千个互连的加速器作为一个紧密耦合的系统进行部署和扩缩。它们非常适合需要极高计算、内存和高速网络同步的工作负载。常见应用场景包括预训练具有数万亿个参数的基础模型、前沿模型部署以及药物发现模拟。

集群化 GPU 的主要特征

  • 同步维护: Google Cloud 同时更新集群中的所有节点,防止单个节点重启导致作业停滞。
  • 自动健康状况监控:系统会主动识别落后节点,并监控硬件故障或静默数据损坏。
  • 主动节点替换:系统会从备用池中逐出性能不佳的虚拟机,并在健康状况良好的机器上重新调度这些虚拟机。
  • 导轨对齐拓扑:导轨对齐拓扑可将 GPU 到 GPU 的流量与标准主机通信隔离开来,从而帮助确保大规模多节点协调的无抖动性能。
  • 高级协议:这些机器系列支持高带宽互连,包括 GPUDirect RDMA(基于 RoCE)。

集群化 GPU 系列和技术规格

请查看下表,了解高级集群化 GPU 机器系列的工作负载和硬件规格。对于百亿亿次级训练,请使用 A4X 或 A3 Ultra 系列。

A4X Max 和 A4X 系列

如果您要训练大规模百亿亿次级基础模型,或运行高度复杂的裸金属高性能计算 (HPC) 模拟,请使用 A4X Max 或 A4X 机器系列。这些系列采用 NVIDIA GB300 Grace Blackwell Ultra 超级芯片,可满足极高的计算和内存需求。

A4X Max(裸金属)

A4X Max 机器类型使用 NVIDIA GB300 Grace Blackwell Ultra 超级芯片 (nvidia-gb300),非常适合基础模型训练和部署。A4X Max 机器类型以裸金属实例的形式提供。

A4X Max 是基于 NVIDIA GB300 NVL72 的艾级平台。每台机器都有两个搭载了 Arm Neoverse V2 核心的 NVIDIA Grace CPU 插槽。这些 CPU 通过快速芯片间 (NVLink-C2C) 通信连接到四个 NVIDIA B300 Blackwell GPU。如需了解机器类型限制,请参阅 A4X Max 和 A4X 限制。

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA GB300 GPU 数量 GPU 内存 (GB HBM3e)
a4x-maxgpu-4g-metal 144 960 12000 6 3600 4 1,116

以下说明适用于上表:

  • vCPU 数量:每个 vCPU 是在其中一个可用的 CPU 平台上以单个硬件超线程的形式实现的。
  • 最大网络带宽:最大出站流量带宽不能超过给定的数字。实际的出站带宽取决于目的地 IP 地址和其他因素。 如需详细了解网络带宽,请参阅网络带宽。
  • GPU 内存:GPU 内存是 GPU 设备上的内存,可用于临时存储数据。它与实例的内存分开,专门用于处理图形密集型工作负载的更高带宽需求。

A4X

A4X 机器类型使用 NVIDIA GB200 Grace Blackwell 超级芯片 (nvidia-gb200),非常适合基础模型训练和部署。

A4X 是基于 NVIDIA GB200 NVL72 的艾级平台。每台机器都有两个搭载了 Arm Neoverse V2 核心的 NVIDIA Grace CPU 插槽。这些 CPU 通过快速芯片间 (NVLink-C2C) 通信连接到四个 NVIDIA B200 Blackwell GPU。如需了解机器类型限制,请参阅 A4X Max 和 A4X 限制。

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA GB200 GPU 数量 GPU 内存 (GB HBM3e)
a4x-highgpu-4g 140 884 12,000 6 2,000 4 744

以下说明适用于上表:

  • vCPU 数量:每个 vCPU 是在其中一个可用的 CPU 平台上以单个硬件超线程的形式实现的。
  • 最大网络带宽:最大出站流量带宽不能超过给定的数字。实际的出站带宽取决于目的地 IP 地址和其他因素。 如需详细了解网络带宽,请参阅网络带宽。
  • GPU 内存:GPU 内存是 GPU 设备上的内存,可用于临时存储数据。它与实例的内存分开,专门用于处理图形密集型工作负载的更高带宽需求。

A4 系列

如果您的目标是训练需要大规模并行处理的前沿基础模型,请使用 A4 机器系列来优化处理时间并最大限度地提高硬件吞吐量。

A4 机器类型挂接了 NVIDIA B200 Blackwell GPU (nvidia-b200),非常适合基础模型训练和部署。如需了解机器类型限制,请参阅 A4 限制。

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA B200 Blackwell GPU 数量 GPU 内存 (GB HBM3e)
a4-highgpu-8g 224 3,968 12,000 10 3600 8 1,440

以下说明适用于上表:

  • vCPU 数量:每个 vCPU 是在其中一个可用的 CPU 平台上以单个硬件超线程的形式实现的。
  • 最大网络带宽:最大出站流量带宽不能超过给定的数字。实际的出站带宽取决于目的地 IP 地址和其他因素。 如需详细了解网络带宽,请参阅网络带宽。
  • GPU 内存:GPU 内存是 GPU 设备上的内存,可用于临时存储数据。它与实例的内存分开,专门用于处理图形密集型工作负载的更高带宽需求。

A3 系列(Ultra、Mega 和 High,配备 8 个 GPU)

如果您需要执行大规模分布式训练运行,或在具有大型数据集表的多个主机上提供前沿模型,请使用 A3 机器系列,以最大限度地减少主机到主机的网络延迟。

A3 Ultra

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA H200 GPU 数量 GPU 内存 (GB HBM3e)
a3-ultragpu-8g 224 2,952 12,000 10 3600 8 1128

A3 Mega

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA H100 GPU 数量 GPU 内存 (GB HBM3)
a3-megagpu-8g 208 1,872 6000 9 1800 8 640

A3 High

机器类型 vCPU 数量 实例内存 (GB) 已挂接的本地 SSD (GiB) 物理 NIC 数量 最大网络带宽 (Gbps) NVIDIA H100 GPU 数量 GPU 内存 (GB HBM3)
a3-highgpu-1g 26 234 750 1 25 1 80
a3-highgpu-2g 52 468 1,500 1 50 2 160
a3-highgpu-4g 104 936 3000 1 100 4 320
a3-highgpu-8g 208 1,872 6000 5 1000 8 640

以下说明适用于上表:

  • vCPU 数量:每个 vCPU 是在其中一个可用的 CPU 平台上以单个硬件超线程的形式实现的。
  • 最大网络带宽:最大出站流量带宽不能超过给定的数字。实际的出站带宽取决于目的地 IP 地址和其他因素。 如需详细了解网络带宽,请参阅网络带宽。
  • GPU 内存:GPU 内存是 GPU 设备上的内存,可用于临时存储数据。它与实例的内存分开,专门用于处理图形密集型工作负载的更高带宽需求。

容量获取

获取集群化 GPU 的计算容量需要进行协调,以管理高性能加速器的供需。由于这些资源实际位于专用网络架构中,因此您必须通过代管式工作流请求这些资源。

您可以通过客户支持团队预留容量,也可以使用未来预留和动态工作负载调度器。

后续步骤