-
Notifications
You must be signed in to change notification settings - Fork 190
/
Copy pathDockerfile.intel_gpu
64 lines (56 loc) · 2.14 KB
/
Dockerfile.intel_gpu
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# The vLLM Dockerfile is used to construct vLLM image that can be directly used
# to run the OpenAI compatible server.
# Based on https://github.com/vllm-project/vllm/blob/main/Dockerfile.openvino
# add Intel ARC support package
FROM ubuntu:22.04 AS dev
RUN apt-get update -y && \
apt-get install -y --no-install-recommends --fix-missing \
ffmpeg \
git \
gnupg2 \
gpg-agent \
libgl1 \
libsm6 \
libxext6 \
python3-pip \
wget
RUN wget -qO - https://repositories.intel.com/gpu/intel-graphics.key | gpg --yes --dearmor --output /usr/share/keyrings/intel-graphics.gpg && \
echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy/lts/2350 unified" | \
tee /etc/apt/sources.list.d/intel-gpu-jammy.list && \
apt-get update -y && apt-get install -y --no-install-recommends --fix-missing \
clinfo \
hwinfo \
intel-level-zero-gpu \
intel-media-va-driver-non-free \
intel-opencl-icd \
level-zero \
libegl1-mesa \
libegl1-mesa-dev \
libegl-mesa0 \
libgbm1 \
libgl1-mesa-dev \
libgl1-mesa-dri \
libglapi-mesa \
libgles2-mesa-dev \
libglx-mesa0 \
libigdgmm12 \
libmfx1 libmfxgen1 \
libvpl2 \
libxatracker2 \
mesa-va-drivers \
mesa-vdpau-drivers \
mesa-vulkan-drivers \
va-driver-all \
vainfo
WORKDIR /workspace
RUN git clone -b v0.6.6.post1 https://github.com/vllm-project/vllm.git
RUN python3 -m pip install -U pip
# install build requirements
RUN PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" python3 -m pip install --no-cache-dir -r /workspace/vllm/requirements-build.txt
# build vLLM with OpenVINO backend
RUN PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" VLLM_TARGET_DEVICE="openvino" python3 -m pip install --no-cache-dir /workspace/vllm/
# fix OpenVINO version to 25.0.0
RUN python3 -m pip install openvino==2025.0.0
#COPY examples/ /workspace/vllm/examples
#COPY benchmarks/ /workspace/vllm/benchmarks
CMD ["/bin/bash"]