NVIDIA/deepops

Tools for building GPU clusters

解决的问题

DeepOps 简化了部署和管理 GPU 服务器集群的复杂过程。它自动化安装必要的软件栈——包括驱动程序、容器运行时和集群管理器——以确保像 NVIDIA DGX 系统这样的硬件能够正确配置,用于 AI 和高性能计算工作负载。

工作原理

DeepOps 使用 Ansible playbook 来编排基础设施的部署。根据用户需求,它可以:

  • 通过 Kubespray 部署完整的 Kubernetes 集群,用于容器化应用管理。
  • 设置 Slurm 集群,用于资源管理和批处理作业调度。
  • 在单台机器上安装必要的 GPU 组件(NVIDIA 驱动程序、Docker 和 NVIDIA 容器运行时)。
  • 在支持的操作系统上部署 NVIDIA DGX 平台软件栈。

适用人群

专为需要在本地或数据中心构建 GPU 支持集群的系统管理员和基础设施工程师设计,特别是使用 NVIDIA DGX 系统或运行 Ubuntu 或 RHEL/Rocky Linux 的通用 GPU 服务器的用户。

主要亮点

  • 多调度器支持:为 Kubernetes 和 Slurm 提供自动化路径。
  • 端到端自动化:涵盖从操作系统级驱动程序到高级集群编排的全部流程。
  • 模块化设计:可用于搭建完整集群,也可仅设置特定组件,如 KubeFlow 或 NFS 存储。
  • DGX 优化:封装了针对 NVIDIA DGX 系统的特定最佳实践。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目