From 06334fae2de7c8a1ea59307b724b992ddf44fb06 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 7 Sep 2026 18:07:27 -0700 Subject: [PATCH] [Build] Remove obsolete TPU Dockerfile (#55376) Signed-off-by: Woosuk Kwon Co-authored-by: OpenAI Codex --- docker/Dockerfile.tpu | 36 ------------------------------------ 1 file changed, 36 deletions(-) delete mode 100644 docker/Dockerfile.tpu diff --git a/docker/Dockerfile.tpu b/docker/Dockerfile.tpu deleted file mode 100644 index ca2d7833c1..0000000000 --- a/docker/Dockerfile.tpu +++ /dev/null @@ -1,36 +0,0 @@ -ARG NIGHTLY_DATE="20250730" -ARG BASE_IMAGE="us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.12_tpuvm_$NIGHTLY_DATE" - -FROM $BASE_IMAGE -WORKDIR /workspace/vllm - -# Install some basic utilities -RUN apt-get update && apt-get install -y \ - git \ - ffmpeg libsm6 libxext6 libgl1 && \ - rm -rf /var/lib/apt/lists/* - -# Build vLLM. -COPY . . -ARG GIT_REPO_CHECK=0 -RUN --mount=type=bind,source=.git,target=.git \ - if [ "$GIT_REPO_CHECK" != 0 ]; then bash tools/check_repo.sh; fi - -# Remove existing versions of dependencies -# TODO: These packages will remain as dead weight in the Docker image layers. -# We should find a way to build the image without uninstalling these. -# Consider using a different base image. -RUN pip uninstall -y torch torch_xla torchvision - -ENV VLLM_TARGET_DEVICE="tpu" -RUN --mount=type=cache,target=/root/.cache/pip \ - --mount=type=bind,source=.git,target=.git \ - python3 -m pip install \ - -r requirements/tpu.txt - -RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install -e . - -# install development dependencies (for testing) -RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install -e tests/vllm_test_utils - -CMD ["/bin/bash"]