[CI/Build] Make the Dockerfile.cpu file's PIP_EXTRA_INDEX_URL Con…

…figurable as a Build Argument (vllm-project#9252)
KuntaiDu · Oct 10, 2024 · 591824a · 591824a
1 parent 7326a7c
commit 591824a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -26,7 +26,8 @@ RUN pip install intel_extension_for_pytorch==2.4.0
 
 WORKDIR /workspace
 
-ENV PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cpu
+ARG PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu"
+ENV PIP_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=bind,src=requirements-build.txt,target=requirements-build.txt \
     pip install --upgrade pip && \