ipex 2.1.40+xpu docker (#4475)

* update stack initial version

* update driver,oneapi stack and validate on rc2

* Update README.md

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 \
+    apt-get install -y --no-install-recommends \
     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} && \
+    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,8 @@ 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 setuptools==69.5.1 \
+        numpy==1.26.4
 
 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/README.md

@@ -34,15 +34,15 @@ export IMAGE_TYPE="xpu"
 To pull docker images use the following command:
 
 ```bash
-docker pull intel/intel-extension-for-pytorch:2.1.40-xpu
+docker pull intel/intel-extension-for-pytorch:2.1.40-xpu-pip-base
 ```
 ### Running container:
 
 Run the following commands to start Intel® Extension for PyTorch\* GPU container. You can use `-v` option to mount your
 local directory into the container. The `-v` argument can be omitted if you do not need
 access to a local directory in the container. 
 
-```
+```bash
 IMAGE_NAME=intel/intel-extension-for-pytorch:2.1.40-xpu
 ```
 ```bash
@@ -98,5 +98,3 @@ Sample output looks like below:
 
 Now you are inside container with Python 3.10, PyTorch, and Intel® Extension for PyTorch\* preinstalled. You can run your own script
 to run on Intel GPU.
-
-

docker/build.sh

@@ -1,24 +1,24 @@
 #!/bin/bash
 
 if [[ ${IMAGE_TYPE} = "xpu" ]]; then
-    IMAGE_NAME=intel/intel-extension-for-pytorch:2.1.40-$IMAGE_TYPE
+    IMAGE_NAME=intel/intel-extension-for-pytorch:2.1.40-$IMAGE_TYPE-pip-base
     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 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 ICD_VER=24.22.29735.27-914~22.04 \
+                 --build-arg LEVEL_ZERO_GPU_VER=1.3.29735.27-914~22.04 \
+                 --build-arg LEVEL_ZERO_VER=1.17.6-914~22.04 \
+                 --build-arg LEVEL_ZERO_DEV_VER=1.17.6-914~22.04 \
+                 --build-arg DPCPP_VER=2024.2.1-1079 \
+                 --build-arg MKL_VER=2024.2.1-103 \
+                 --build-arg CCL_VER=2021.13.1-31 \
+                 --build-arg TORCH_VERSION=2.1.0.post3+cxx11.abi  \
                  --build-arg IPEX_VERSION=2.1.40+xpu \
-                 --build-arg TORCHVISION_VERSION=0.16.0.post2+cxx11.abi \
-                 --build-arg TORCHAUDIO_VERSION=2.1.0.post2+cxx11.abi \
+                 --build-arg TORCHVISION_VERSION=0.16.0.post3+cxx11.abi \
+                 --build-arg TORCHAUDIO_VERSION=2.1.0.post3+cxx11.abi \
                  --build-arg ONECCL_BIND_PT_VERSION=2.1.400+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/ \
@@ -28,4 +28,3 @@ if [[ ${IMAGE_TYPE} = "xpu" ]]; then
                  -t ${IMAGE_NAME} \
                  -f Dockerfile.prebuilt .
 fi
-