Installation with XPU

vLLM initially supports basic model inferencing and serving on Intel GPU platform.

Table of contents:

  1. Requirements

  2. Quick start using Dockerfile

  3. Build from source

Requirements

  • OS: Linux

  • Supported Hardware: Intel Data Center GPU (Intel ARC GPU WIP)

  • OneAPI requirements: oneAPI 2024.1

Quick start using Dockerfile

  1. $ docker build -f Dockerfile.xpu -t vllm-xpu-env --shm-size=4g .
  2. $ docker run -it \
  3. --rm \
  4. --network=host \
  5. --device /dev/dri \
  6. -v /dev/dri/by-path:/dev/dri/by-path \
  7. vllm-xpu-env

Build from source

  • First, install required driver and intel OneAPI 2024.1 or later.

  • Second, install Python packages for vLLM XPU backend building:

  1. $ source /opt/intel/oneapi/setvars.sh
  2. $ pip install --upgrade pip
  3. $ pip install -v -r requirements-xpu.txt
  • Finally, build and install vLLM XPU backend:
  1. $ VLLM_TARGET_DEVICE=xpu python setup.py install

Note

  • FP16 is the default data type in the current XPU backend. The BF16 data type will be supported in the future.