vLLM CPU: Fix Triton Version to Resolve Related Error(#12893)
This commit is contained in:
parent
ab3fc66eb7
commit
ae9f5320da
1 changed files with 1 additions and 1 deletions
|
|
@ -100,7 +100,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
pip uninstall -y intel-extension-for-pytorch && \
|
pip uninstall -y 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 triton==3.1.0 && \
|
||||||
# Clean up unnecessary files to reduce image size
|
# Clean up unnecessary files to reduce image size
|
||||||
rm -rf /var/lib/apt/lists/*
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue