From 9ab326713fccd8acf1b0a8fd9cca157f39c20688 Mon Sep 17 00:00:00 2001 From: Richard Liu <39319471+richardsliu@users.noreply.github.com> Date: Tue, 11 Mar 2025 08:01:35 -0700 Subject: [PATCH] Uninstall dependencies before installing requirements/tpu.txt (#14586) Signed-off-by: Signed-off-by: Richard Liu --- Dockerfile.tpu | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Dockerfile.tpu b/Dockerfile.tpu index 960dc8e9..50806d88 100644 --- a/Dockerfile.tpu +++ b/Dockerfile.tpu @@ -15,6 +15,9 @@ 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 +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 \