vLLM: Fix vLLM-CPU docker image (#12741)

This commit is contained in:
Xiangyu Tian 2025-01-24 10:00:29 +08:00 committed by GitHub
parent 69f13c78b8
commit f924880694
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -28,6 +28,7 @@ RUN wget -qO /sbin/tini https://github.com/krallin/tini/releases/download/${TINI
cd ./vllm && \
git checkout v0.6.6.post1 && \
pip install cmake>=3.26 wheel packaging ninja "setuptools-scm>=8" numpy && \
pip uninstall intel-extension-for-pytorch && \
pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu && \
VLLM_TARGET_DEVICE=cpu python3 setup.py install && \
pip install ray