Skip to content

Commit

Permalink
adding reference to rocm compatible image
Browse files Browse the repository at this point in the history
  • Loading branch information
Fiona-Waters authored and openshift-merge-bot[bot] committed Sep 17, 2024
1 parent bc38ed5 commit 6e895a8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/cluster-configuration.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ cluster = Cluster(ClusterConfiguration(
labels={"exampleLabel": "example", "secondLabel": "example"},
))
```
Note: 'quay.io/modh/ray:2.35.0-py39-cu121' is the default image used by the CodeFlare SDK for creating a RayCluster resource. If you have your own Ray image which suits your purposes, specify it in image field to override the default image.
Note: 'quay.io/modh/ray:2.35.0-py39-cu121' is the default image used by the CodeFlare SDK for creating a RayCluster resource. If you have your own Ray image which suits your purposes, specify it in image field to override the default image. If you are using ROCm compatible GPUs you can use 'quay.io/modh/ray:2.35.0-py39-rocm61'. You can also find documentation on building a custom image [here](https://github.com/opendatahub-io/distributed-workloads/tree/main/images/runtime/examples).

The `labels={"exampleLabel": "example"}` parameter can be used to apply additional labels to the RayCluster resource.

Expand Down

0 comments on commit 6e895a8

Please sign in to comment.