vLLM: Fix vLLM-CPU docker image (#12741)
This commit is contained in:
parent
69f13c78b8
commit
f924880694
1 changed files with 1 additions and 0 deletions
|
|
@ -28,6 +28,7 @@ RUN wget -qO /sbin/tini https://github.com/krallin/tini/releases/download/${TINI
|
||||||
cd ./vllm && \
|
cd ./vllm && \
|
||||||
git checkout v0.6.6.post1 && \
|
git checkout v0.6.6.post1 && \
|
||||||
pip install cmake>=3.26 wheel packaging ninja "setuptools-scm>=8" numpy && \
|
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 && \
|
pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu && \
|
||||||
VLLM_TARGET_DEVICE=cpu python3 setup.py install && \
|
VLLM_TARGET_DEVICE=cpu python3 setup.py install && \
|
||||||
pip install ray
|
pip install ray
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue