Add Ray cluster spec for TPU pods #1292
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
kevin85421
reviewed
Aug 7, 2023
| annotations: | ||
| {} | ||
| labels: | ||
| cloud.google.com/gke-ray-node-type: head |
There was a problem hiding this comment.
Is this a special label for TPU?
There was a problem hiding this comment.
These are for TPU telemetry - allows us to help identify TPU node issues for customers. It's not strictly needed but is useful for support purposes.
| imagePullSecrets: | ||
| [] | ||
| containers: | ||
| - volumeMounts: |
There was a problem hiding this comment.
Should we add ports info explicitly?
There was a problem hiding this comment.
Do you mean the TPU/JAX ports? I'm not sure if they are needed.
There was a problem hiding this comment.
I mean the ports for the dashboard and GCS.
kuberay/ray-operator/config/samples/ray-cluster.complete.yaml
Lines 35 to 38 in 4f85055
kevin85421
approved these changes
Aug 10, 2023
blublinsky
pushed a commit
to blublinsky/kuberay
that referenced
this pull request
Aug 15, 2023
Add Ray cluster spec for TPU pods
lowang-bh
pushed a commit
to lowang-bh/kuberay
that referenced
this pull request
Sep 24, 2023
Add Ray cluster spec for TPU pods
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This is a sample Kuberay spec for deploying a Ray cluster on a 2x2x2 TPU v4 topology.
For TPU onboarding (prerequisite), please refer to the documentation here: https://docs.google.com/document/d/1TRwtfi2pzXbT6We0WQdwNzDdKHruk3u1XeLLJwW38kE/edit#heading=h.8ntu2hqwqvhl