提交智能体收藏0
技术动态

NVIDIA FLARE 支持 Docker、Kubernetes 与 Slurm 跨平台联邦学习

NVIDIA 开发者博客介绍 FLARE 的双层架构:持久化联邦服务与按需启动的任务工作进程分离,使同一联邦内各站点可分别使用 Docker、Kubernetes 或 Slurm 执行后端,并通过 study 实现逻辑多租户隔离。

本次变化

据 NVIDIA 开发者博客 2026 年 9 月 15 日的文章,NVIDIA FLARE 采用两层架构:长期运行的 server/client 父进程负责维护联邦、认证连接与协调工作;独立的 server/client 任务工作进程负责执行提交的联邦学习任务。父进程无需占用 GPU,任务提交后由各站点按配置的执行平台启动工作进程,用完资源后退出。

文章称 FLARE 2.8 引入 Docker 与 Kubernetes 部署支持,2.9 增加 Slurm 支持。站点可分别选择:Docker 用于单主机容器化执行;Kubernetes 通过 Helm 安装父 pod,按调度器放置任务 pod;Slurm 将每个工作进程作为批作业提交,支持单节点或多节点分配及裸执行、Pyxis/Enroot、Apptainer 容器。

study 与资源共享

study 在同一部署内提供逻辑多租户边界,定义参与的客户端站点和可开启会话的管理员用户。站点通过本地 study_runtime.yaml 把 study 映射到数据集挂载、环境变量、Secret 引用、默认镜像以及 Kubernetes pod 模板、Docker 运行时设置、Slurm 分区/账户/QoS 等策略。任务通过 meta.json 中的 resource_spec 以可移植方式声明 GPU、整数 CPU 单元和内存需求,各启动器再翻译为 Docker、Kubernetes 或 Slurm 的原生参数。文章强调 study 适用于共用同一 FLARE 服务器与 PKI、但需要逻辑隔离的实验;需要独立 PKI、独立管理员或独立故障域的应使用独立部署。

限制与待核实

本文未提供实际部署命令、性能数据或测试结果,也未验证异构集群下的具体表现;Docker/Kubernetes/Slurm 支持对应版本以原文为准,本地资源与安全策略仍由各站点自行配置和验证。

来源与核对说明

  1. NVIDIA 开发者官方博客 ↗https://developer.nvidia.com/blog/scaling-federated-learning-across-docker-kubernetes-and-slurm-with-nvidia-flare/

本文经 AI 整理及独立自动校对,未做运行验证;自动校对仍可能遗漏问题,请以原始资料为准。

这篇内容对你有帮助吗?

点赞与收藏保存在当前浏览器。

继续阅读

NVIDIA 将 Run:ai 平台开源并纳入 DSX AI Factory →

NVIDIA cuOpt 推出 mPDLP 多 GPU 线性规划求解器 →

NVIDIA 在 CUDA 13.1 中通过 Runtime API 开放 Green Contexts →

NVIDIA 发布 AICR v1.0:为 GPU 集群配置提供开放、稳定、可验证的契约 →