From 792595b59d01eb7060b6cbc6dec20c129ea917b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Lucchesi?= Date: Tue, 29 Apr 2025 19:36:48 +0200 Subject: [PATCH] [TPU][V1][CI] Replace `python3 setup.py develop` with standard `pip install --e` on TPU (#17374) Signed-off-by: NickLucche --- docker/Dockerfile.tpu | 2 +- .../getting_started/installation/ai_accelerator/tpu.inc.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.tpu b/docker/Dockerfile.tpu index 50806d8820a30..295270d29f765 100644 --- a/docker/Dockerfile.tpu +++ b/docker/Dockerfile.tpu @@ -23,7 +23,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \ --mount=type=bind,source=.git,target=.git \ python3 -m pip install \ -r requirements/tpu.txt -RUN python3 setup.py develop +RUN python3 -m pip install -e . # install development dependencies (for testing) RUN python3 -m pip install -e tests/vllm_test_utils diff --git a/docs/source/getting_started/installation/ai_accelerator/tpu.inc.md b/docs/source/getting_started/installation/ai_accelerator/tpu.inc.md index 8beb92ef7da0a..4459cc61e1cde 100644 --- a/docs/source/getting_started/installation/ai_accelerator/tpu.inc.md +++ b/docs/source/getting_started/installation/ai_accelerator/tpu.inc.md @@ -158,7 +158,7 @@ sudo apt-get install libopenblas-base libopenmpi-dev libomp-dev Run the setup script: ```bash -VLLM_TARGET_DEVICE="tpu" python setup.py develop +VLLM_TARGET_DEVICE="tpu" python -m pip install -e . ``` ## Set up using Docker