We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent af295e9 commit 9ab3267Copy full SHA for 9ab3267
Dockerfile.tpu
@@ -15,6 +15,9 @@ ARG GIT_REPO_CHECK=0
15
RUN --mount=type=bind,source=.git,target=.git \
16
if [ "$GIT_REPO_CHECK" != 0 ]; then bash tools/check_repo.sh; fi
17
18
+# Remove existing versions of dependencies
19
+RUN pip uninstall -y torch torch_xla torchvision
20
+
21
ENV VLLM_TARGET_DEVICE="tpu"
22
RUN --mount=type=cache,target=/root/.cache/pip \
23
--mount=type=bind,source=.git,target=.git \
0 commit comments