From 637ed06fbf151dab22068dbce33396b4d32a5f79 Mon Sep 17 00:00:00 2001
From: Alexei-V-Ivanov-AMD
 <156011006+Alexei-V-Ivanov-AMD@users.noreply.github.com>
Date: Mon, 18 Nov 2024 17:07:07 -0600
Subject: [PATCH] Update Dockerfile.rocm

Restoring functionality of 'docker build' after pytorch version labels have been pruned.
---
 Dockerfile.rocm | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 8fb79afaebe97..f8ef3932dea78 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -51,9 +51,9 @@ RUN --mount=type=cache,target=/root/.cache/pip \
         *"rocm-6.2"*) \
             python3 -m pip uninstall -y torch torchvision \
             && python3 -m pip install --pre \
-                torch==2.6.0.dev20240918 \
+                torch==2.6.0.20241113 \
                 'setuptools-scm>=8' \
-                torchvision==0.20.0.dev20240918 \
+                torchvision==0.20.0.dev20241113+rocm6.2 \
                 --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2;; \
         *) ;; esac