Skip to content

Commit 9ab3267

Browse files
authored
Uninstall dependencies before installing requirements/tpu.txt (vllm-project#14586)
Signed-off-by: <[email protected]> Signed-off-by: Richard Liu <[email protected]>
1 parent af295e9 commit 9ab3267

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Dockerfile.tpu

+3
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)