update stack for 2.3.110+xpu (#4726)

Author: sramakintel

docker/Dockerfile.prebuilt

@@ -23,7 +23,7 @@ ARG DEBIAN_FRONTEND=noninteractive
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends --fix-missing \
+    apt-get install -y --no-install-recommends \
         apt-utils \
         build-essential \
         ca-certificates \
@@ -40,8 +40,8 @@ RUN apt-get update && \
     rm -rf  /var/lib/apt/lists/*
 
 RUN wget -qO - https://repositories.intel.com/gpu/intel-graphics.key | \
-    gpg --dearmor --output /usr/share/keyrings/intel-graphics.gpg
-RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy/lts/2350 unified" | \
+    gpg --dearmor --yes --output /usr/share/keyrings/intel-graphics.gpg
+RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy unified" | \
     tee /etc/apt/sources.list.d/intel-gpu-jammy.list
 
 ARG ICD_VER
@@ -50,11 +50,11 @@ ARG LEVEL_ZERO_VER
 ARG LEVEL_ZERO_DEV_VER
 
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends --fix-missing \
-    intel-opencl-icd=${ICD_VER} \
-    intel-level-zero-gpu=${LEVEL_ZERO_GPU_VER} \
-    level-zero=${LEVEL_ZERO_VER} \
-    level-zero-dev=${LEVEL_ZERO_DEV_VER} && \
+    apt-get install -y --no-install-recommends \
+        intel-opencl-icd=${ICD_VER} \
+        intel-level-zero-gpu=${LEVEL_ZERO_GPU_VER} \
+        libze1=${LEVEL_ZERO_VER} \
+        libze-dev=${LEVEL_ZERO_DEV_VER} && \
     apt-get clean && \
     rm -rf  /var/lib/apt/lists/*
 
@@ -68,7 +68,7 @@ ARG MKL_VER
 ARG CCL_VER
 
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends --fix-missing \
+    apt-get install -y --no-install-recommends \
         intel-oneapi-runtime-dpcpp-cpp=${DPCPP_VER} \
         intel-oneapi-runtime-mkl=${MKL_VER} \
         intel-oneapi-runtime-ccl=${CCL_VER} && \
@@ -80,10 +80,10 @@ RUN apt-get update && apt install -y software-properties-common
 RUN add-apt-repository -y ppa:deadsnakes/ppa 
 
 RUN apt-cache policy $PYTHON && apt-get update && apt-get install -y \
-    --no-install-recommends --fix-missing $PYTHON 
+    --no-install-recommends $PYTHON 
 
-RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
-    ${PYTHON} lib${PYTHON} python3-pip ${PYTHON}-distutils && \
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        ${PYTHON} lib${PYTHON} python3-pip ${PYTHON}-distutils && \
     apt-get clean && \
     rm -rf  /var/lib/apt/lists/*
 
@@ -107,7 +107,7 @@ ARG TORCHVISION_WHL_URL
 ARG TORCHAUDIO_WHL_URL
 ARG ONECCL_BIND_PT_WHL_URL
 
-RUN python -m pip install numpy 
+RUN python -m pip install numpy
 
 RUN python -m pip install torch==${TORCH_VERSION} --extra-index-url ${TORCH_WHL_URL} && \
     python -m pip install intel_extension_for_pytorch==${IPEX_VERSION} --extra-index-url ${IPEX_WHL_URL} && \

docker/build.sh

@@ -1,25 +1,25 @@
 #!/bin/bash
 
 if [[ ${IMAGE_TYPE} = "xpu" ]]; then
-    IMAGE_NAME=intel/intel-extension-for-pytorch:2.1.30-$IMAGE_TYPE
+    IMAGE_NAME=intel/intel-extension-for-pytorch:2.3.110-$IMAGE_TYPE
     docker build --build-arg http_proxy=$http_proxy \
                  --build-arg https_proxy=$https_proxy \
                  --build-arg no_proxy=" " \
                  --build-arg NO_PROXY=" " \
                  --build-arg UBUNTU_VERSION=22.04 \
                  --build-arg PYTHON=python3.10 \
-                 --build-arg ICD_VER=23.43.27642.40-803~22.04 \
-                 --build-arg LEVEL_ZERO_GPU_VER=1.3.27642.40-803~22.04 \
-                 --build-arg LEVEL_ZERO_VER=1.14.0-744~22.04 \
-                 --build-arg LEVEL_ZERO_DEV_VER=1.14.0-744~22.04 \
+                 --build-arg ICD_VER=24.26.30049.10-950~22.04 \
+                 --build-arg LEVEL_ZERO_GPU_VER=1.3.30049.10-950~22.04 \
+                 --build-arg LEVEL_ZERO_VER=1.17.6-950~22.04 \
+                 --build-arg LEVEL_ZERO_DEV_VER=1.17.6-950~22.04 \
                  --build-arg DPCPP_VER=2024.1.0-963 \
                  --build-arg MKL_VER=2024.1.0-691 \
                  --build-arg CCL_VER=2021.12.0-309 \
-                 --build-arg TORCH_VERSION=2.1.0.post2+cxx11.abi  \
-                 --build-arg IPEX_VERSION=2.1.30+xpu \
-                 --build-arg TORCHVISION_VERSION=0.16.0.post2+cxx11.abi \
-                 --build-arg TORCHAUDIO_VERSION=2.1.0.post2+cxx11.abi \
-                 --build-arg ONECCL_BIND_PT_VERSION=2.1.300+xpu \
+                 --build-arg TORCH_VERSION=2.3.1+cxx11.abi  \
+                 --build-arg IPEX_VERSION=2.3.110+xpu \
+                 --build-arg TORCHVISION_VERSION=0.18.1+cxx11 \
+                 --build-arg TORCHAUDIO_VERSION=2.3.1+cxx11 \
+                 --build-arg ONECCL_BIND_PT_VERSION=2.3.100+xpu \
                  --build-arg TORCH_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
                  --build-arg IPEX_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
                  --build-arg TORCHVISION_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \