Skip to content

Commit 3aec49e

Browse files
authored
[ci/build] update nightly torch for gh200 test (vllm-project#12270)
Signed-off-by: youkaichao <[email protected]>
1 parent c646128 commit 3aec49e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ WORKDIR /workspace
5252
# after this step
5353
RUN --mount=type=cache,target=/root/.cache/pip \
5454
if [ "$TARGETPLATFORM" = "linux/arm64" ]; then \
55-
python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu124 "torch==2.6.0.dev20241210+cu124" "torchvision==0.22.0.dev20241215"; \
55+
python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu126 "torch==2.7.0.dev20250121+cu126" "torchvision==0.22.0.dev20250121"; \
5656
fi
5757

5858
COPY requirements-common.txt requirements-common.txt

0 commit comments

Comments
 (0)