Skip to content

Commit 6e895a8

Browse files
adding reference to rocm compatible image
1 parent bc38ed5 commit 6e895a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/cluster-configuration.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ cluster = Cluster(ClusterConfiguration(
2323
labels={"exampleLabel": "example", "secondLabel": "example"},
2424
))
2525
```
26-
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.
26+
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).
2727

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

0 commit comments

Comments
 (0)