From f7e2591f15ea3f135ca0c89fd5f3711cdc55ed66 Mon Sep 17 00:00:00 2001
From: Ziteng Zhang <87107332+Jasonzzt@users.noreply.github.com>
Date: Fri, 23 Feb 2024 15:02:08 +0800
Subject: [PATCH] [LLM] change IPEX230 to IPEX220 in dockerfile (#10222)

* change IPEX230 to IPEX220 in dockerfile
---
 docker/llm/inference/cpu/docker/Dockerfile | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/docker/llm/inference/cpu/docker/Dockerfile b/docker/llm/inference/cpu/docker/Dockerfile
index d422d03d..c8cb047f 100644
--- a/docker/llm/inference/cpu/docker/Dockerfile
+++ b/docker/llm/inference/cpu/docker/Dockerfile
@@ -61,16 +61,15 @@ RUN cp /root/miniconda3/condabin/conda /usr/bin && \
     . ~/.bashrc && \
     conda activate bigdl-speculative-py39 && \
     cd /llm && \
-    pip3 install --no-cache-dir --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu && \
     pip install --pre --upgrade bigdl-llm[all] && \
-    wget https://raw.githubusercontent.com/intel/intel-extension-for-pytorch/main/scripts/compile_bundle.sh && \
-    bash compile_bundle.sh && \
+    pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cpu && \
+    pip install intel-extension-for-pytorch==2.2.0 && \
+    pip install oneccl_bind_pt==2.2.0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/ && \
     cd ./intel-extension-for-pytorch && \
     pip install -r requirements.txt && \
     pip install transformers==4.36.2 && \
     pip install transformers_stream_generator && \
     cd /llm && \
-    rm -rf llvm-project llvm-release intel-extension-for-pytorch compile_bundle.sh && \
     echo "conda deactivate" >> /root/.bashrc
     
-ENTRYPOINT ["/bin/bash"]
\ No newline at end of file
+ENTRYPOINT ["/bin/bash"]