Skip to content

GitLab

Unverified Commit 3aec49e5 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[ci/build] update nightly torch for gh200 test (#12270) 


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent c6461280
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
Dockerfile
View file @ 3aec49e5
...@@ -52,7 +52,7 @@ WORKDIR /workspace ...@@ -52,7 +52,7 @@ WORKDIR /workspace
# after this step # after this step
RUN --mount=type=cache,target=/root/.cache/pip \ RUN --mount=type=cache,target=/root/.cache/pip \
if [ "$TARGETPLATFORM" = "linux/arm64" ]; then \ if [ "$TARGETPLATFORM" = "linux/arm64" ]; then \
python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu124 "torch==2.6.0.dev20241210+cu124" "torchvision==0.22.0.dev20241215"; \ python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu126 "torch==2.7.0.dev20250121+cu126" "torchvision==0.22.0.dev20250121"; \
fi fi
COPY requirements-common.txt requirements-common.txt COPY requirements-common.txt requirements-common.txt
......
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