Skip to content

GitLab

Unverified Commit c16eaac5 authored by Jie Fu (傅杰)'s avatar Jie Fu (傅杰) Committed by GitHub
Browse files

[Hardware][Intel CPU] Update torch 2.4.0 for CPU backend (#6931)

parent db351863
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Dockerfile.cpu
View file @ c16eaac5
...@@ -26,7 +26,7 @@ COPY ./ /workspace/vllm ...@@ -26,7 +26,7 @@ COPY ./ /workspace/vllm
WORKDIR /workspace/vllm WORKDIR /workspace/vllm
RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/test/cpu RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
# Support for building with non-AVX512 vLLM: docker build --build-arg VLLM_CPU_DISABLE_AVX512="true" ... # Support for building with non-AVX512 vLLM: docker build --build-arg VLLM_CPU_DISABLE_AVX512="true" ...
ARG VLLM_CPU_DISABLE_AVX512 ARG VLLM_CPU_DISABLE_AVX512
......
requirements-cpu.txt
View file @ c16eaac5
...@@ -2,5 +2,5 @@ ...@@ -2,5 +2,5 @@
-r requirements-common.txt -r requirements-common.txt
# Dependencies for x86_64 CPUs # Dependencies for x86_64 CPUs
torch == 2.4.0; platform_machine != "ppc64le" torch == 2.4.0+cpu; platform_machine != "ppc64le"
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch
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