Unverified Commit 9ab32671 authored by Richard Liu's avatar Richard Liu Committed by GitHub
Browse files

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



Signed-off-by: <ricliu@google.com>
Signed-off-by: default avatarRichard Liu <ricliu@google.com>
parent af295e9b
......@@ -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 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment