From ca3d0a9d02c9d322b76320c4b6f8fef1be079eb0 Mon Sep 17 00:00:00 2001 From: Shilpa Chugh Date: Thu, 6 Feb 2025 11:49:04 +0530 Subject: [PATCH] Update TrainingROCm Image --- support/defaults.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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" )