[Docker] Release docker gpu (#2503)

intel · Nov 20, 2023 · c90b1b2 · c90b1b2
1 parent c560519
commit c90b1b2
Show file tree

Hide file tree

Showing 3 changed files with 662 additions and 100 deletions.
diff --git a/docker/build.sh b/docker/build.sh
@@ -21,13 +21,13 @@ if [ $IMAGE_TYPE == "xpu" -o $IMAGE_TYPE == "gpu" ]
 then
         IMAGE_NAME=intel-extension-for-tensorflow:$IMAGE_TYPE
         docker build --no-cache --build-arg UBUNTU_VERSION=22.04 \
-                                --build-arg ICD_VER=23.17.26241.33-647~22.04 \
-                                --build-arg LEVEL_ZERO_GPU_VER=1.3.26241.33-647~22.04 \
-                                --build-arg LEVEL_ZERO_VER=1.11.0-647~22.04 \
-                                --build-arg LEVEL_ZERO_DEV_VER=1.11.0-647~22.04 \
-                                --build-arg DPCPP_VER=2023.2.1-16 \
-                                --build-arg MKL_VER=2023.2.0-49495 \
-                                --build-arg CCL_VER=2021.10.0-49084 \
+                                --build-arg ICD_VER=23.30.26918.50-736~22.04 \
+                                --build-arg LEVEL_ZERO_GPU_VER=1.3.26918.50-736~22.04 \
+                                --build-arg LEVEL_ZERO_VER=1.13.1-719~22.04 \
+                                --build-arg LEVEL_ZERO_DEV_VER=1.13.1-719~22.04 \
+                                --build-arg DPCPP_VER=2024.0.0-49819 \
+                                --build-arg MKL_VER=2024.0.0-49656 \
+                                --build-arg CCL_VER=2021.11.0-49156 \
                                 --build-arg PYTHON=python3.10 \
                                 --build-arg TF_VER=2.14 \
                                 --build-arg WHEELS=*.whl \

diff --git a/docker/itex-xpu.Dockerfile b/docker/itex-xpu.Dockerfile
@@ -39,9 +39,9 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf  /var/lib/apt/lists/*
 
-RUN wget -qO - https://repositories.intel.com/graphics/intel-graphics.key | \
+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/graphics/ubuntu jammy max" | \
+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
@@ -94,7 +94,7 @@ RUN ln -sf $(which ${PYTHON}) /usr/local/bin/python && \
     ln -sf $(which ${PYTHON}) /usr/bin/python && \
     ln -sf $(which ${PYTHON}) /usr/bin/python3
 
-ARG TF_VER="2.13"
+ARG TF_VER="2.14"
 
 RUN pip --no-cache-dir install tensorflow==${TF_VER}