Uninstall dependencies before installing requirements/tpu.txt (#14586)

Signed-off-by: <[email protected]> Signed-off-by: Richard Liu <[email protected]>
vllm-project · Mar 11, 2025 · 9ab3267 · 9ab3267
1 parent af295e9
commit 9ab3267
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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 \