Skip to content

Commit c748082

Browse files
committed
adding reference to rocm compatible image
1 parent d77d5eb commit c748082

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/cluster-configuration.md

Lines changed: 1 addition & 1 deletion
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'.
2727

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

0 commit comments

Comments
 (0)