From 9bf81a05e1f062543941bf1d08f4fc75fb328325 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= <dtrifiro@redhat.com>
Date: Tue, 7 Jan 2025 16:46:36 +0100
Subject: [PATCH] Dockerfile.rocm.ubi: bump torch/torchvision to 20250104 build

---
 Dockerfile.rocm.ubi | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.rocm.ubi b/Dockerfile.rocm.ubi
index 8766b995bb555..da3959b32a807 100644
--- a/Dockerfile.rocm.ubi
+++ b/Dockerfile.rocm.ubi
@@ -49,8 +49,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
     uv pip install --pre \
         --index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
-        torch==2.6.0.dev20241107+rocm${version}\
-        torchvision==0.20.0.dev20241107+rocm${version} && \
+        torch==2.6.0.dev20250104+rocm${version}\
+        torchvision==0.20.0.dev20250104+rocm${version} && \
     # Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
     microdnf install -y libdrm-amdgpu && \
     microdnf clean all