Skip to content

Commit 9d20e5f

Browse files
committed
Revert "Reapply "Install pre-built xformers-0.0.32.post2 built with pt-2.9.0" (vllm-project#27768)"
This reverts commit ba33e88.
1 parent df33486 commit 9d20e5f

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

docker/Dockerfile

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,13 @@ RUN --mount=type=bind,from=build,src=/workspace/dist,target=/vllm-workspace/dist
361361
&& uv pip install --system dist/*.whl --verbose \
362362
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.')
363363

364+
# TODO (huydhn): Remove this once xformers is released for 2.9.0
365+
RUN --mount=type=cache,target=/root/.cache/uv bash - <<'BASH'
366+
. /etc/environment
367+
export TORCH_CUDA_ARCH_LIST='7.5 8.0+PTX 9.0a'
368+
uv pip install --system --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.32.post2"
369+
BASH
370+
364371
# Install FlashInfer pre-compiled kernel cache and binaries
365372
# https://docs.flashinfer.ai/installation.html
366373
RUN --mount=type=cache,target=/root/.cache/uv \

requirements/cuda.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ torch==2.9.0
99
torchaudio==2.9.0
1010
# These must be updated alongside torch
1111
torchvision==0.24.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
12-
# Build from https://github.com/facebookresearch/xformers/releases/tag/v0.0.32.post1
13-
xformers==0.0.33+5d4b92a5.d20251029; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.9
12+
# https://github.com/facebookresearch/xformers/releases/tag/v0.0.32.post1
13+
# xformers==0.0.32.post1; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.8
1414
# FlashInfer should be updated together with the Dockerfile
1515
flashinfer-python==0.4.1

0 commit comments

Comments
 (0)