diff --git a/docker/llm/serving/xpu/docker/Dockerfile b/docker/llm/serving/xpu/docker/Dockerfile index dd3979200fa..ae40611b387 100644 --- a/docker/llm/serving/xpu/docker/Dockerfile +++ b/docker/llm/serving/xpu/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM intel/oneapi-basekit:2024.1.1-devel-ubuntu22.04 +FROM intel/oneapi-basekit:2024.2.1-0-devel-ubuntu22.04 ARG http_proxy ARG https_proxy @@ -39,7 +39,7 @@ RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRO python3 get-pip.py && \ rm get-pip.py && \ pip install --upgrade requests argparse urllib3 && \ - pip install --pre --upgrade ipex-llm[xpu,serving] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ && \ + pip install --pre --upgrade ipex-llm[xpu-arc,serving] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ && \ pip install transformers_stream_generator einops tiktoken && \ pip install --upgrade colorama && \ # Download all-in-one benchmark and examples @@ -62,11 +62,11 @@ RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRO rm -rf ./ipex-llm && \ # Install torch-ccl cd /tmp/ && \ - pip install torch==2.1.0.post2 torchvision==0.16.0.post2 torchaudio==2.1.0.post2 intel-extension-for-pytorch==2.1.30.post0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ && \ + # pip install torch==2.1.0.post2 torchvision==0.16.0.post2 torchaudio==2.1.0.post2 intel-extension-for-pytorch==2.1.30.post0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ && \ # Internal oneccl wget https://sourceforge.net/projects/oneccl-wks/files/2024.0.0.6.2-release/oneccl_wks_installer_2024.0.0.6.2.sh && \ bash oneccl_wks_installer_2024.0.0.6.2.sh && \ - git clone https://github.com/intel/torch-ccl -b v2.1.300+xpu && \ + git clone https://github.com/intel/torch-ccl -b v2.3.100+xpu && \ cd torch-ccl && \ patch -p1 < /tmp/oneccl-binding.patch && \ USE_SYSTEM_ONECCL=ON COMPUTE_BACKEND=dpcpp python setup.py install && \