当前位置：首页 > news >正文

dstack 是 Kubernetes 和 Slurm 的开源替代方案，旨在简化 ML 团队跨顶级云、本地集群和加速器的 GPU 分配和 AI 工作负载编排

news 2025/5/5 21:13:37

一、软件介绍

文末提供程序和源码下载

dstack 是 Kubernetes 和 Slurm 的开源替代方案，旨在简化顶级云和本地集群中 ML 团队的 GPU 分配和 AI 工作负载编排。

二、Accelerators 加速器

dstack 支持 NVIDIA 开箱即用的、 AMD 、 Google TPU 和 Intel Gaudi 加速器。

三、Installation 安装

在通过 CLI 或 API 使用 dstack 之前，请设置一个 dstack 服务器。如果您已经有一个正在运行的 dstack 服务器，则只需设置 CLI。

Set up the server 设置服务器

（可选）配置后端

要与云提供商一起使用 dstack ，请通过该文件 ~/.dstack/server/config.yml 配置后端。

Start the server 启动服务器
You can install the server on Linux, macOS, and Windows (via WSL 2). It requires Git and OpenSSH.
您可以在 Linux、macOS 和 Windows 上安装服务器（通过 WSL 2）。它需要 Git 和 OpenSSH。

pip 果仁
$ pip install "dstack[all]" -U
uv
$ uv tool install "dstack[all]" -U
Once it's installed, go ahead and start the server.
安装后，继续启动服务器。

$ dstack server
Applying ~/.dstack/server/config.yml...

The admin token is "bbae0f28-d3dd-4820-bf61-8f4bb40815da"
The server is running at http://127.0.0.1:3000/
For more details on server configuration options, see the Server deployment guide.
有关 Server 配置选项的更多详细信息，请参阅 Server 部署指南。

Set up the CLI 设置 CLI
Once the server is up, you can access it via the dstack CLI.
服务器启动后，您可以通过 dstack CLI 访问它。

The CLI can be installed on Linux, macOS, and Windows. It requires Git and OpenSSH.
CLI 可以安装在 Linux、macOS 和 Windows 上。它需要 Git 和 OpenSSH。

pip 果仁
$ pip install dstack -U
uv
$ uv tool install dstack -U
To point the CLI to the dstack server, configure it with the server address, user token, and project name:
要将 CLI 指向服务器， dstack 请使用服务器地址、用户令牌和项目名称对其进行配置：

$ dstack config \
--url http://127.0.0.1:3000 \
--project main \
--token bbae0f28-d3dd-4820-bf61-8f4bb40815da

Configuration is updated at ~/.dstack/config.yml