2024-06-28 21:50:16 +08:00
|
|
|
# The vLLM Dockerfile is used to construct vLLM image that can be directly used
|
|
|
|
|
# to run the OpenAI compatible server.
|
|
|
|
|
|
2024-07-31 02:33:01 +08:00
|
|
|
FROM ubuntu:22.04 AS dev
|
2024-06-28 21:50:16 +08:00
|
|
|
|
|
|
|
|
RUN apt-get update -y && \
|
2024-09-24 00:44:26 +08:00
|
|
|
apt-get install -y \
|
|
|
|
|
git python3-pip \
|
|
|
|
|
ffmpeg libsm6 libxext6 libgl1
|
2024-06-28 21:50:16 +08:00
|
|
|
WORKDIR /workspace
|
|
|
|
|
|
2024-10-15 02:34:47 +08:00
|
|
|
COPY . .
|
2024-06-28 21:50:16 +08:00
|
|
|
|
|
|
|
|
# install build requirements
|
|
|
|
|
RUN PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" python3 -m pip install -r /workspace/vllm/requirements-build.txt
|
|
|
|
|
# build vLLM with OpenVINO backend
|
2024-08-08 00:49:10 +08:00
|
|
|
RUN PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" VLLM_TARGET_DEVICE="openvino" python3 -m pip install /workspace/vllm/
|
2024-06-28 21:50:16 +08:00
|
|
|
|
|
|
|
|
COPY examples/ /workspace/vllm/examples
|
|
|
|
|
COPY benchmarks/ /workspace/vllm/benchmarks
|
|
|
|
|
|
|
|
|
|
CMD ["/bin/bash"]
|