Skip to content

Commit 38e0f85

Browse files
richardsliuJC1DA
authored andcommitted
Uninstall dependencies before installing requirements/tpu.txt (vllm-project#14586)
Signed-off-by: <ricliu@google.com> Signed-off-by: Richard Liu <ricliu@google.com>
1 parent a118752 commit 38e0f85

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Dockerfile.tpu

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,9 @@ ARG GIT_REPO_CHECK=0
1515
RUN --mount=type=bind,source=.git,target=.git \
1616
if [ "$GIT_REPO_CHECK" != 0 ]; then bash tools/check_repo.sh; fi
1717

18+
# Remove existing versions of dependencies
19+
RUN pip uninstall -y torch torch_xla torchvision
20+
1821
ENV VLLM_TARGET_DEVICE="tpu"
1922
RUN --mount=type=cache,target=/root/.cache/pip \
2023
--mount=type=bind,source=.git,target=.git \

0 commit comments

Comments
 (0)