ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update BigDL-LLM-inference image (#9805)

Browse source 
* upgrade to oneapi 2024

* Pin level-zero-gpu version

* add flag
This commit is contained in:
Guancheng Fu 2024-01-03 14:00:09 +08:00 committed by GitHub
parent 5c6543e070
commit 0396fafed1
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
docker/llm/inference/xpu/docker/Dockerfile
View file

@ -1,4 +1,4 @@
FROM intel/oneapi-basekit:2023.2.1-devel-ubuntu22.04
FROM intel/oneapi-basekit:2024.0.1-devel-ubuntu22.04
ARG http_proxy
ARG https_proxy
@ -35,10 +35,10 @@ RUN curl -fsSL https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-P
python3 get-pip.py && \
rm get-pip.py && \
pip install --upgrade requests argparse urllib3 && \
pip install --pre --upgrade bigdl-llm[xpu] -f https://developer.intel.com/ipex-whl-stable-xpu && \
pip install --pre --upgrade bigdl-llm[xpu_2.1] -f https://developer.intel.com/ipex-whl-stable-xpu && \
# Install opencl-related repos
apt-get update && \
apt-get install -y intel-opencl-icd intel-level-zero-gpu level-zero level-zero-dev && \
apt-get install -y intel-opencl-icd intel-level-zero-gpu=1.3.26241.33-647~22.04 level-zero level-zero-dev --allow-downgrades && \
# Install related libary of chat.py
pip install --upgrade colorama && \
# Install vllm dependencies