Skip to content

Commit c6264ba

Browse files
change triton to 3.4.0
1 parent c0a8e3d commit c6264ba

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/arm64/gpu/Dockerfile.arm64

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,8 @@ RUN uv pip install --no-cache-dir -U \
5050
${TORCHVISION_URL} \
5151
${TORCHAUDIO_URL}
5252

53-
RUN uv pip install pytorch_triton==3.3.0 \
54-
--extra-index-url https://download.pytorch.org/whl/cu128 --torch-backend=cu128
53+
RUN uv pip install pytorch_triton==3.4.0 \
54+
--extra-index-url https://download.pytorch.org/whl/pytorch-triton/
5555

5656
##############################################################################
5757
FROM base AS build-base

0 commit comments

Comments
 (0)