[CI/Build] Update Dockerfile.rocm (vllm-project#10434)

Signed-off-by: Alexei V. Ivanov <alexei.ivanov@amd.com>
liuyanyi · Nov 20, 2024 · d5b68ab · d5b68ab
1 parent a324d3a
commit d5b68ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.dev20241113+rocm6.2 \
                 '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