Dockerfile.rocm.ubi: bump torch version to 2.5.0.dev20240912+rocm6.1

opendatahub-io · Sep 25, 2024 · 69ac6c1 · 69ac6c1
1 parent c29c9f4
commit 69ac6c1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Dockerfile.rocm.ubi b/Dockerfile.rocm.ubi
@@ -49,9 +49,8 @@ RUN microdnf -y install \
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=cache,target=/root/.cache/uv \
     uv pip install -v --index-url "https://download.pytorch.org/whl/nightly/rocm6.1" \
-        torch==2.5.0.dev20240726+rocm6.1 \
-        torchvision==0.20.0.dev20240726+rocm6.1
-
+        torch==2.5.0.dev20240912+rocm6.1 \
+        torchvision==0.20.0.dev20240912+rocm6.1
 
 FROM rocm_base as rocm_devel