Skip to content

GitLab

Unverified Commit d5b68aba authored by Alexei-V-Ivanov-AMD's avatar Alexei-V-Ivanov-AMD Committed by GitHub
Browse files

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


Signed-off-by: default avatarAlexei V. Ivanov <alexei.ivanov@amd.com>
parent a324d3a1
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Dockerfile.rocm
View file @ d5b68aba
...@@ -51,9 +51,9 @@ RUN --mount=type=cache,target=/root/.cache/pip \ ...@@ -51,9 +51,9 @@ RUN --mount=type=cache,target=/root/.cache/pip \
*"rocm-6.2"*) \ *"rocm-6.2"*) \
python3 -m pip uninstall -y torch torchvision \ python3 -m pip uninstall -y torch torchvision \
&& python3 -m pip install --pre \ && python3 -m pip install --pre \
torch==2.6.0.dev20240918 \ torch==2.6.0.dev20241113+rocm6.2 \
'setuptools-scm>=8' \ '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;; \ --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2;; \
*) ;; esac *) ;; esac
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment