diff --git a/support/defaults.go b/support/defaults.go index 099e5f1..5cedae1 100644 --- a/support/defaults.go +++ b/support/defaults.go @@ -11,5 +11,5 @@ const ( RayTorchCudaImage = "quay.io/rhoai/ray:2.35.0-py311-cu121-torch24-fa26" RayTorchROCmImage = "quay.io/rhoai/ray:2.35.0-py311-rocm61-torch24-fa26" TrainingCudaImage = "quay.io/modh/training:py311-cuda121-torch241" - TrainingROCmImage = "quay.io/modh/training:py311-rocm61-torch241" + TrainingROCmImage = "quay.io/modh/training:py311-rocm62-torch241" )