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

Signed-off-by: <ricliu@google.com>
Signed-off-by: Richard Liu <ricliu@google.com>
This commit is contained in:
Richard Liu 2025-03-11 08:01:35 -07:00 committed by GitHub
parent af295e9b01
commit 9ab326713f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 \