← 文章 / AI技术
NVIDIA 开发者博客 11天前 · 2026-10-24 06:37:20 · 11 阅读

跨 8 台 NVIDIA DGX Spark 节点运行前沿级 AI 模型

AI 开发者不再局限于运行单一模型,而是并行编排多个模型、串联推理调用,并试验需要数百 GB 显存的尖端架构。NVIDIA DGX Spark 旨在将 AI 开发能力带到桌面端,而现在,你可以连接更多的 Spark 节点组成集群。

在 GTC Berlin 大会上,NVIDIA 宣布正式支持 8 节点 DGX Spark 集群,统一内存扩展至 1 TB,为尖端模型、多智能体工作负载以及接近线性的预训练本地运行打开大门。

NVIDIA Sync 让组集群变得简单,其 Cluster Assistant 负责处理配置与设备管理。NVIDIA Sync 对 8 节点集群的支持预计年底前上线。

继续阅读,了解随集群规模扩大,推理、编排与训练性能的基准测试数据。

运行更大、更新的尖端模型

8 节点集群的最大优势在于内存。单台 DGX Spark 提供 128 GB 统一内存;4 节点达到 512 GB;8 节点则达到 1 TB。这一容量让那些此前在 4 节点系统上无法启动的模型成为可能,例如 Z.ai 的 GLM 5.3 (753B) 完整模型。

除了单纯容纳这些模型,该集群的设计目标还包括交互式服务。以早期案例为例,我们在 8 节点集群上运行了 GLM 5.3 Flash,测得生成速度约为每秒 80 个 tokens(TPS),足以支持对尖端混合专家(MoE)模型进行交互式本地开发。

GLM 5.3 (753B) 完整检查点可以在 8x 128 GB 集群上运行,成为你本地可服务的最大规模开源模型之一。

并发量预填充,总 tokens/秒(TPS)预填充,每用户(TPS)解码,总(TPS)解码,每用户(TPS)
11,6001,6003131
41,6007505121
81,6004736718
表 1. 完整 GLM 5.3(nvidia/GLM-5.3-NVFP4)在八节点 DGX Spark 集群上的性能表现。该模型 NVFP4 格式约 470 GB,使用 MTP3 在并发数为 1、4、8 时测得。Prefill 按 8K 输入序列长度(ISL)测试。

使用 Spark 优化后端加速集合通信

张量并行推理会花费相当一部分时间在 all-reduce 等集合通信操作上。在通过交换机互联的集群上,我们一直在评估一个针对 Spark 优化的集合通信后端 RoCEnante,它能降低小消息、低并发、中等 rank 数量场景下的集合通信延迟——这正是本地 AI 硬件上张量并行 decode 的典型模式。

该后端先执行 all-to-all 交换,再进行本地归约。每个 rank 将自己的输入分发给所有其他 rank,然后在本地完成求和,整个过程由单个 GPU kernel 协调完成。

主要优势是减少了通信步数。Ring all-reduce 需要在相邻 rank 之间多轮传递数据,而这种做法直接与所有对端交换数据,省去了逐级转发的链条。在小消息和中等数量 rank 的场景下,可以降低延迟。

代价是更多的数据搬运。每个 rank 都要向所有对端发送完整输入,并自行归约全部输入。随着消息体或 rank 数量(节点数)的增长,这部分额外流量开销会越来越大,Ring all-reduce 在带宽效率上的优势也就愈发明显。如果需要支持高并发、更大规模的集群或 ring 拓扑,NVIDIA Collective Communications Library(NCCL)会有更好的表现。

折线图展示了随着 Spark 节点数从 1 增加到 8,解码速度(相对于单节点 Spark)的提升情况。图中绘制了两条线:“启用 ROCEnante”和“未启用 ROCEnante”。两者在单节点时均起始于 1.00 倍。在 4 个节点时,启用 ROCEnante 达到 3.06 倍,未启用为 2.56 倍。在 8 个节点时,启用 ROCEnante 达到 4.43 倍,未启用为 3.05 倍。配置:交换拓扑,Qwen3.8-27B-NVFP4,禁用推测解码。
图 1. 在交换拓扑下使用 Qwen3.8-27B-NVFP4(禁用推测解码)时,Spark 解码性能随节点数的扩展情况,分别对应启用和未启用 ROCEnante。在 8 个 Spark 节点时,启用 ROCEnante 可实现 4.43 倍的加速,而未启用仅为 3.05 倍——扩展效率提升了 45%。

实际测试中,在 4 节点(NVFP4)下,Prefill 和 Decode 吞吐量分别提升约 5% 和 21%;而在 8 节点的非量化模型下,Decode 吞吐量提升约 18%。这与上述的延迟与带宽权衡相符。该后端已集成至 FlashInfer,若启动测试失败,会透明地回退至 NCCL。

该后端仅适用于交换连接拓扑,因为它依赖于节点间的直接点对点交换。它并非 NCCL 的替代品,NCCL 仍是受支持的标准默认选项,且随着集群规模和并发量的增加,NCCL 是更佳选择。最初的 One-shot All-reduce 方案由社区成员(Jason Cook/el8)贡献。

二、四、八:集群扩展

低延迟的 NVIDIA ConnectX-7 网卡 结合基于融合以太网的 RDMA(RoCE)通信,使 DGX Spark 节点能够高效交换张量。因此,你可以将模型扩展到更多节点,以突破内存限制并提升吞吐量。为了隔离集群扩展行为并消除上下文依赖影响,我们在单节点到八节点之间,对未使用推测解码/MTP 的稠密 Qwen3.8-27B(NVFP4)模型进行了基准测试:

Spark 节点数拓扑结构Prefill (TPS)Prefill 扩展倍率Decode (TPS)Decode 扩展倍率
1—2,6351.00x11.861.00x
2直连3,2851.25x21.711.838x
4物理环4,4461.69x32.822.77x
4交换机 (ROCEnante)4,5561.73x36.343.06x
8物理环4,9621.88x40.113.38x
8交换机 (ROCEnante)5,1051.94x52.584.43x
表 2. NVIDIA/Qwen3.8-27B (NVFP4) 在不同拓扑下的扩展表现,支持时使用 ROCEnante 后端。测试工具为 llama-benchy,ISL 8192 tokens,OSL 256 tokens,并发数为 1。扩展倍率相对于相同精度下单个 Spark 节点。

有两点趋势尤为显著。首先,Decode 吞吐量随节点数量显著增加,从单节点的约 12 TPS 提升至八节点的 52 TPS。

其次,最佳拓扑取决于工作负载:若需将单个模型分散到所有节点,物理环拓扑能提供出色的 Decode 吞吐量。但一旦需要将集群拆分为独立组,或运行要求节点间任意直接连通的工作负载,就必须使用交换机。

若想自行尝试,可以参考这份网络指南,它详细讲解了从双节点配置起步并逐步扩展集群设置的过程。

无交换机环状拓扑与设置简便性

并非所有集群都需要交换机。DGX Spark 节点可以直接连成环形,既降低成本又简化部署,对爱好者和中小规模部署来说很有吸引力。

在四节点 DGX Spark 部署中,NCCL 针对张量并行通信默认采用 ring 算法,无交换机的环形拓扑正好与流量走向完全吻合,能以更低成本提供可观的吞吐。如表 2 所示,四节点环形拓扑在该工作负载的 prefill 和 decode 吞吐上表现良好。

但一旦需要扩展规模或承担专门任务,情况就不同了:

  • 八节点或拆分分配: 如果想把集群拆成比如两组四节点,环形拓扑无法表达这种结构,必须用交换机。
  • All-to-all 和传统 HPC 工作负载: 任何依赖点对点通信模式的工作负载(例如基于 MPI 的集群)都需要交换机,环形拓扑会付出额外的跳数开销。
  • 优化后的 collective 通信后端: 上述针对 Spark 优化的后端要求所有节点之间直接点对点互联,因此只适用于交换机连接的配置。

简单来说:如果你只想要最简单、最便宜的四节点推理服务器,选无交换机环形拓扑;如果需要扩展到八节点、拆分集群,或者运行环形拓扑无法支持的通信模式,就选交换机。

用 Switchyard 做多模型编排

并不是每个请求都需要动用最大的模型。在一个共享的八节点集群上,把简单请求路由给小而快的模型、把大模型留给难题,可以大幅提升整体吞吐。

NVIDIA NeMo Switchyard 让这一方案落地:一个轻量级分类器检查每个请求(或会话),并将其路由到合适的模型池,同时由负载均衡器在多个副本间分配流量。由于这种布局把八个节点拆分成相互独立的模型池,Switchyard 需要运行在交换机连接的集群上,因为跨副本路由请求需要任意节点间的全连通能力。

在原型中,我们采用 OpenBMB 的 MiniCPM5 2B 作为简单任务的分类与执行模型,复杂任务则交给 Qwen3.8 Flash Next。在一个涵盖标准、进阶及高难度用例的工具调用基准测试中,该路由代理在处理质量上与仅使用 Qwen 的配置相当,同时释放了集群的容量。

在包含 92 个用例(含高难度案例)的工具调用基准测试中,路由代理得分为 91/100,与仅使用 Qwen 的配置持平。

由于每个副本中 MiniCPM 的每秒请求量约为 Qwen 的一个数量级,将部分流量分流至更小的模型可以提升整个集群的吞吐量上限。在 8 个节点上分配 3 个 Qwen 副本(每个 Spark 节点 2 个)和 2 个 MiniCPM 副本(每个 Spark 节点 1 个),可得:

  • 当一半请求路由至 Qwen 时,相较于仅使用 Qwen 的基线,整体吞吐量提升 1.5 倍。
  • 当四分之一请求路由至 Qwen,其余路由至 MiniCPM 时,整体吞吐量提升 3.0 倍。

路由策略是一个简单的 YAML 文件,NVIDIA 提供了示例策略作为起点,开发者可根据自身工作负载调整分流比例和分类器。

从零开始预训练:基于 nanochat 的近线性扩展

训练行为与推理不同。当每个节点持有模型全量副本时,节点间唯一的通信是每步一次的梯度同步,因此集群扩展接近线性。为证明这一点,我们从零预训练了一个 nanochat 模型(约 896.5M 参数,20 层 Transformer,隐藏维度 1,280,序列长度 2,048 个 token),并测量了随节点数增加的吞吐量。

节点数TPS加速比扩展效率
18,7411.00x100.0%
217,2021.97x98.4%
433,6403.85x96.2%
868,0797.79x97.4%
表 3. Nanochat 预训练吞吐量从单节点扩展到八节点 DGX Spark 的性能表现(全局批处理每步 524,288 tokens,每 GPU 微批次 32 条序列)

八节点集群实现了 7.79 倍的加速比,效率高达 97.4%。这种接近线性的扩展能力,使得八节点 Spark 集群成为从头训练小型模型的可行平台,而不仅仅是用于部署大模型。它同样是一个自然的进阶工作流:先在本地完成原型设计和预训练,随着模型规模增长再横向扩展。

如需复现 Nanochat 预训练实验或将其适配到自有模型,完整代码已在 GitHub 上公开。

开启多系统 DGX Spark 之旅

八节点 DGX Spark 集群能在桌面级空间内提供加载前沿级模型的内存、同时编排多个模型的灵活性,以及从头训练小型模型的高效性。

了解更多详情,请查看:

立即在 NVIDIA DGX Spark 上开始构建。

原始来源: NVIDIA 开发者博客

评论 (0)