开始使用 Ray#

本页介绍如何在单机或集群上启动 Ray。

小技巧

在按照本页面的说明操作之前,请确保已 安装 Ray

什么是 Ray 运行时?#

Ray 程序能够通过利用底层的 Ray 运行时 来实现并行化和分布式处理。Ray 运行时由多个在后台启动的服务/进程组成,用于通信、数据传输、调度等。Ray 运行时可以在笔记本电脑、单个服务器或多个服务器上启动。

启动 Ray 运行时有三种方式:

在所有情况下,ray.init() 都会尝试自动找到一个 Ray 实例进行连接。它会按顺序检查:1. RAY_ADDRESS 操作系统环境变量。2. 传递给 ray.init(address=<address>) 的具体地址。3. 如果没有提供地址,则检查使用 ray start 在同一台机器上启动的最新 Ray 实例。

在单台机器上启动 Ray#

调用 ray.init() 会在你的笔记本电脑/机器上启动一个本地 Ray 实例。这台笔记本电脑/机器成为“头节点”。

备注

在Ray的最新版本(>=1.5)中,ray.init() 将在首次使用Ray远程API时自动调用。

import ray
# Other Ray APIs will not work until `ray.init()` is called.
ray.init()
import io.ray.api.Ray;

public class MyRayApp {

  public static void main(String[] args) {
    // Other Ray APIs will not work until `Ray.init()` is called.
    Ray.init();
    ...
  }
}
#include <ray/api.h>
// Other Ray APIs will not work until `ray::Init()` is called.
ray::Init()

当调用 ray.init() 的进程终止时,Ray 运行时也会终止。要显式停止或重启 Ray,请使用 shutdown API。

import ray
ray.init()
... # ray program
ray.shutdown()
import io.ray.api.Ray;

public class MyRayApp {

  public static void main(String[] args) {
    Ray.init();
    ... // ray program
    Ray.shutdown();
  }
}
#include <ray/api.h>
ray::Init()
... // ray program
ray::Shutdown()

要检查 Ray 是否已初始化,请使用 is_initialized API。

import ray
ray.init()
assert ray.is_initialized()

ray.shutdown()
assert not ray.is_initialized()
import io.ray.api.Ray;

public class MyRayApp {

public static void main(String[] args) {
        Ray.init();
        Assert.assertTrue(Ray.isInitialized());
        Ray.shutdown();
        Assert.assertFalse(Ray.isInitialized());
    }
}
#include <ray/api.h>

int main(int argc, char **argv) {
    ray::Init();
    assert(ray::IsInitialized());

    ray::Shutdown();
    assert(!ray::IsInitialized());
}

请参阅 配置 文档,了解配置 Ray 的各种方式。

通过CLI启动Ray(ray start#

使用 CLI 中的 ray start 在一台机器上启动一个单节点的 Ray 运行时。这台机器成为“头节点”。

$ ray start --head --port=6379

Local node IP: 192.123.1.123
2020-09-20 10:38:54,193 INFO services.py:1166 -- View the Ray dashboard at http://localhost:8265

--------------------
Ray runtime started.
--------------------

...

你可以通过在运行 ray start 的同一节点上启动一个驱动程序进程来连接到这个 Ray 实例。ray.init() 现在会自动连接到最新的 Ray 实例。

import ray
ray.init()
import io.ray.api.Ray;

public class MyRayApp {

  public static void main(String[] args) {
    Ray.init();
    ...
  }
}
java -classpath <classpath> \
  -Dray.address=<address> \
  <classname> <args>
#include <ray/api.h>

int main(int argc, char **argv) {
  ray::Init();
  ...
}
RAY_ADDRESS=<address> ./<binary> <args>

你可以将其他节点连接到头节点,通过在这些节点上调用 ray start 来创建一个 Ray 集群。更多详情请参见 启动一个本地集群。在集群的任何机器上调用 ray.init() 将连接到同一个 Ray 集群。

启动一个 Ray 集群 (ray up)#

Ray 集群可以通过 集群启动器 启动。ray up 命令使用 Ray 集群启动器在云上启动一个集群,创建一个指定的“头节点”和工作节点。在底层,它会自动调用 ray start 来创建一个 Ray 集群。

你的代码**只**需要在集群中的一台机器上执行(通常是头节点)。了解更多关于 在Ray集群上运行程序 的信息。

要连接到 Ray 集群,请从集群中的某台机器调用 ray.init。这将连接到最新的 Ray 集群:

ray.init()

需要注意的是,调用 ray up 的机器不会被视为 Ray 集群的一部分,因此在同一台机器上调用 ray.init 不会连接到集群。

接下来是什么?#

查看我们的 部署部分 以获取更多关于在不同环境中部署 Ray 的信息,包括 KubernetesYARNSLURM