You can leverage your Kubernetes cluster as a substrate for execution of distributed Ray programs. The Ray Autoscaler<cluster-index>
spins up and deletes Kubernetes Pods according to the resource demands of the Ray workload. Each Ray node runs in its own Kubernetes Pod.
This document cover the following topics:
Intro to the Ray Kubernetes Operator<ray-operator>
Launching Ray clusters with the Ray Helm Chart<ray-helm>
Monitoring Ray clusters<ray-k8s-monitor>
Running Ray programs using Ray Client<ray-k8s-client>
You can find more information at the following links:
Ray Operator and Helm chart configuration<k8s-advanced>
GPU usage with Kubernetes<k8s-gpus>
Using Ray Tune on your Kubernetes cluster<tune-kubernetes>
How to manually set up a non-autoscaling Ray cluster on Kubernetes<ray-k8s-static>
Deployments of Ray on Kubernetes are managed by the Ray Kubernetes Operator
. The Ray Operator follows the standard Kubernetes Operator pattern. The main players are
- A Custom Resource called a
RayCluster
, which describes the desired state of the Ray cluster. - A Custom Controller, the
Ray Operator
, which processesRayCluster
resources and manages the Ray cluster.
Under the hood, the Operator uses the Ray Autoscaler<cluster-index>
to launch and scale your Ray cluster.
The rest of this document explains how to launch a small example Ray cluster on Kubernetes.
Ray on Kubernetes Configuration and Advanced Usage<k8s-advanced>
.
Ray provides a Helm chart to simplify deployment of the Ray Operator and Ray clusters.
The Ray Helm chart is available as part of the Ray GitHub repository. The chart will be published to a public Helm repository as part of a future Ray release.
- Configure kubectl to access your Kubernetes cluster.
- Install Helm 3.
- Download the Ray Helm chart.
To run the default example in this document, make sure your Kubernetes cluster can accomodate additional resource requests of 4 CPU and 2.5Gi memory.
You can install a small Ray cluster with a single helm
command. The default cluster configuration consists of a Ray head pod and two worker pods, with scaling allowed up to three workers.
# Navigate to the directory containing the chart
$ cd ray/deploy/charts
# Install a small Ray cluster with the default configuration
# in a new namespace called "ray". Let's name the Helm release "example-cluster."
$ helm -n ray install example-cluster --create-namespace ./ray
NAME: example-cluster
LAST DEPLOYED: Fri May 14 11:44:06 2021
NAMESPACE: ray
STATUS: deployed
REVISION: 1
TEST SUITE: None
View the installed resources as follows.
# The custom resource representing the state of the Ray cluster.
$ kubectl -n ray get rayclusters
NAME STATUS RESTARTS AGE
example-cluster Running 0 53s
# The Ray head node and two Ray worker nodes.
$ kubectl -n ray get pods
NAME READY STATUS RESTARTS AGE
example-cluster-ray-head-type-5926k 1/1 Running 0 57s
example-cluster-ray-worker-type-8gbwx 1/1 Running 0 40s
example-cluster-ray-worker-type-l6cvx 1/1 Running 0 40s
# A service exposing the Ray head node.
$ kubectl -n ray get service
NAME TYPE CLUSTER-IP EXTERNAL-IP PORT(S) AGE
example-cluster-ray-head ClusterIP 10.8.11.17 <none> 10001/TCP,8265/TCP,8000/TCP 115s
# The operator deployment.
# By default, the deployment is launched in namespace "default".
$ kubectl get deployment ray-operator
NAME READY UP-TO-DATE AVAILABLE AGE
ray-operator 1/1 1 1 3m1s
# The single pod of the operator deployment.
$ kubectl get pod -l cluster.ray.io/component=operator
NAME READY STATUS RESTARTS AGE
ray-operator-84f5d57b7f-xkvtm 1/1 Running 0 3m35
# The Custom Resource Definition defining a RayCluster.
$ kubectl get crd rayclusters.cluster.ray.io
NAME CREATED AT
rayclusters.cluster.ray.io 2021-05-14T18:44:02
To view autoscaling logs, run a kubectl logs
command on the operator pod:
# The last 100 lines of logs.
$ kubectl logs \
$(kubectl get pod -l cluster.ray.io/component=operator -o custom-columns=:metadata.name) \
| tail -n 100
The Ray dashboard<ray-dashboard>
can be accessed on the Ray head node at port 8265
.
# Forward the relevant port from the service exposing the Ray head.
$ kubectl -n ray port-forward service/example-cluster-ray-head 8265:8265
# The dashboard can now be viewed in a browser at http://localhost:8265
Ray Job Submission <jobs-overview>
can be used to submit Ray programs to your Ray cluster.
export RAY_ADDRESS="http://127.0.0.1:8265"
❯ ray job submit --runtime-env-json='{"working_dir": "./", "pip": ["requests==2.26.0"]}' -- "python script.py"
2021-12-01 23:04:52,672 INFO cli.py:25 -- Creating JobSubmissionClient at address: http://127.0.0.1:8265
2021-12-01 23:04:52,809 INFO sdk.py:144 -- Uploading package gcs://_ray_pkg_bbcc8ca7e83b4dc0.zip.
2021-12-01 23:04:52,810 INFO packaging.py:352 -- Creating a file package for local directory './'.
2021-12-01 23:04:52,878 INFO cli.py:105 -- Job submitted successfully: raysubmit_RXhvSyEPbxhcXtm6.
2021-12-01 23:04:52,878 INFO cli.py:106 -- Query the status of the job using: `ray job status raysubmit_RXhvSyEPbxhcXtm6`.
Ray Client <ray-client>
can be used to interactively execute Ray programs on your Ray cluster. The Ray Client server runs on the Ray head node, on port 10001
.
Note
Connecting with Ray client requires using matching minor versions of Python (for example 3.7) on the server and client end, that is, on the Ray head node and in the environment where ray.init("ray://<host>:<port>")
is invoked. Note that the default rayproject/ray
images use Python 3.7. The latest offical Ray release builds are available for Python 3.6 and 3.8 at the Ray Docker Hub.
Connecting with Ray client also requires matching Ray versions. To connect from a local machine to a cluster running the examples in this document, the latest release version<installation>
of Ray must be installed locally.
One way to connect to the Ray cluster from outside your Kubernetes cluster is to forward the Ray Client server port:
$ kubectl -n ray port-forward service/example-cluster-ray-head 10001:10001
Then open a new shell and try out a sample Ray program:
$ python ray/doc/kubernetes/example_scripts/run_local_example.py
The program in this example uses ray.init("ray://127.0.0.1:10001")
to connect to the Ray cluster. The program waits for three Ray nodes to connect and then tests object transfer between the nodes.
You can also connect to your Ray cluster from another pod in the same Kubernetes cluster.
For example, you can submit a Ray application to run on the Kubernetes cluster as a Kubernetes Job. The Job will run a single pod running the Ray driver program to completion, then terminate the pod but allow you to access the logs.
The following command submits a Job which executes an example Ray program.
$ kubectl -n ray create -f https://raw.githubusercontent.com/ray-project/ray/master/doc/kubernetes/job-example.yaml
job.batch/ray-test-job created
The program executed by the job uses the name of the Ray cluster's head Service to connect: ray.init("ray://example-cluster-ray-head:10001")
. The program waits for three Ray nodes to connect and then tests object transfer between the nodes.
To view the output of the Job, first find the name of the pod that ran it, then fetch its logs:
$ kubectl -n ray get pods
NAME READY STATUS RESTARTS AGE
example-cluster-ray-head-type-5926k 1/1 Running 0 21m
example-cluster-ray-worker-type-8gbwx 1/1 Running 0 21m
example-cluster-ray-worker-type-l6cvx 1/1 Running 0 21m
ray-test-job-dl9fv 1/1 Running 0 3s
# Fetch the logs. You should see repeated output for 10 iterations and then
# 'Success!'
$ kubectl -n ray logs ray-test-job-dl9fv
# Cleanup
$ kubectl -n ray delete job ray-test-job
job.batch "ray-test-job" deleted
Tip
Code dependencies for a given Ray task or actor must be installed on each Ray node that might run the task or actor. Typically, this means that all Ray nodes need to have the same dependencies installed. To achieve this, you can build a custom container image, using one of the official Ray images as the base. Alternatively, try out the experimental Runtime Environments<runtime-environments>
API (latest Ray release version recommended.)
To remove a Ray Helm release and the associated API resources, use kubectl delete and helm uninstall. Note the order of the commands below.
# First, delete the RayCluster custom resource.
$ kubectl -n ray delete raycluster example-cluster
raycluster.cluster.ray.io "example-cluster" deleted
# Delete the Ray release.
$ helm -n ray uninstall example-cluster
release "example-cluster" uninstalled
# Optionally, delete the namespace created for our Ray release.
$ kubectl delete namespace ray
namespace "ray" deleted
Note that helm uninstall
does not delete the RayCluster CRD. If you wish to delete the CRD, make sure all Ray Helm releases have been uninstalled, then run kubectl delete crd rayclusters.cluster.ray.io
.
More details on resource cleanup<k8s-cleanup>
Ray Operator Advanced Configuration<k8s-advanced>