Dockerfile.rocm.ubi: bump torch/torchvision to 20250104 build

opendatahub-io · Jan 7, 2025 · 9bf81a0 · 9bf81a0
1 parent 39bd136
commit 9bf81a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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