Skip to content

Commit c16eaac

Browse files
authored
[Hardware][Intel CPU] Update torch 2.4.0 for CPU backend (vllm-project#6931)
1 parent db35186 commit c16eaac

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile.cpu

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ COPY ./ /workspace/vllm
2626

2727
WORKDIR /workspace/vllm
2828

29-
RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/test/cpu
29+
RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
3030

3131
# Support for building with non-AVX512 vLLM: docker build --build-arg VLLM_CPU_DISABLE_AVX512="true" ...
3232
ARG VLLM_CPU_DISABLE_AVX512

requirements-cpu.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@
22
-r requirements-common.txt
33

44
# Dependencies for x86_64 CPUs
5-
torch == 2.4.0; platform_machine != "ppc64le"
5+
torch == 2.4.0+cpu; platform_machine != "ppc64le"
66
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch

0 commit comments

Comments
 (0)