Skip to content

Commit

Permalink
[Docker] update stack versions for itex 2.15.0.1 release (#2720)
Browse files Browse the repository at this point in the history
  • Loading branch information
zufangzhu authored Aug 8, 2024
1 parent ffa38b7 commit 18806f1
Show file tree
Hide file tree
Showing 6 changed files with 51 additions and 15 deletions.
4 changes: 2 additions & 2 deletions docker/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,13 @@ wget <download link from https://pypi.org/project/intel-optimization-for-horovod
### I. Customize Build Script
We provide [build.sh](./build.sh) as the Docker container build script. The OS version and some software versions (such as Python and TensorFlow) are hard coded inside the script. If you're using a different version, you can edit this script.

For example, to build a Docker container with Python 3.10 and TensorFlow 2.15 on an Ubuntu 22.04 layer, update [build.sh](./build.sh) as shown below.
For example, to build a Docker container with Python 3.10 and TensorFlow 2.15.1 on an Ubuntu 22.04 layer, update [build.sh](./build.sh) as shown below.

```bash
IMAGE_NAME=intel-extension-for-tensorflow:cpu-ubuntu
docker build --build-arg UBUNTU_VERSION=22.04 \
--build-arg PYTHON=python3.10 \
--build-arg TF_VER=2.15 \
--build-arg TF_VER=2.15.1 \
--build-arg WHEELS=*.whl \
-t $IMAGE_NAME \
-f itex-cpu.Dockerfile .
Expand Down
14 changes: 7 additions & 7 deletions docker/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,23 +21,23 @@ 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.43.27642.40-803~22.04 \
--build-arg LEVEL_ZERO_GPU_VER=1.3.27642.40-803~22.04 \
--build-arg ICD_VER=23.43.27642.50-803~22.04 \
--build-arg LEVEL_ZERO_GPU_VER=1.3.27642.50-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 DPCPP_VER=2024.2.1-1079 \
--build-arg MKL_VER=2024.2.1-103 \
--build-arg CCL_VER=2021.13.1-31 \
--build-arg PYTHON=python3.10 \
--build-arg TF_VER=2.15 \
--build-arg TF_VER=2.15.1 \
--build-arg WHEELS=*.whl \
-t $IMAGE_NAME \
-f itex-xpu.Dockerfile .
else
IMAGE_NAME=intel-extension-for-tensorflow:$IMAGE_TYPE
docker build --no-cache --build-arg UBUNTU_VERSION=22.04 \
--build-arg PYTHON=python3.10 \
--build-arg TF_VER=2.15 \
--build-arg TF_VER=2.15.1 \
--build-arg WHEELS=*.whl \
-t $IMAGE_NAME \
-f itex-cpu.Dockerfile .
Expand Down
2 changes: 1 addition & 1 deletion docker/itex-cpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ RUN pip --no-cache-dir install --upgrade \
pip \
setuptools

ARG TF_VER="2.15"
ARG TF_VER="2.15.1"

RUN pip --no-cache-dir install tensorflow==${TF_VER}

Expand Down
2 changes: 1 addition & 1 deletion docker/itex-xpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ RUN pip --no-cache-dir install --upgrade \
pip \
setuptools

ARG TF_VER="2.15"
ARG TF_VER="2.15.1"

RUN pip --no-cache-dir install tensorflow==${TF_VER}

Expand Down
44 changes: 40 additions & 4 deletions tools/python/config.json
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@

{
"latest_release": "2.15.0.0",
"latest_release": "2.15.0.1",
"os_list" : [
"ubuntu",
"rhel",
Expand All @@ -15,6 +15,7 @@
"2.13.0.0" : "20.04 22.04",
"2.14.0.1" : "22.04",
"2.15.0.0" : "22.04",
"2.15.0.1" : "22.04",
"latest" : "22.04"
},

Expand All @@ -25,6 +26,7 @@
"2.13.0.0" : "8.6",
"2.14.0.1" : "8.6 8.8 9.0 9.2",
"2.15.0.0" : "8.6 8.8 8.9 9.0 9.2 9.3",
"2.15.0.1" : "8.6 8.8 8.9 9.0 9.2 9.3",
"latest" : "8.6 8.8 8.9 9.0 9.2 9.3"
},

Expand All @@ -35,6 +37,7 @@
"2.13.0.0" : "15.3 15.4",
"2.14.0.1" : "15.4 15.5",
"2.15.0.0" : "15.4 15.5",
"2.15.0.1" : "15.4 15.5",
"latest" : "15.4 15.5"
}
},
Expand All @@ -46,6 +49,7 @@
"2.13.0.0" : 8,
"2.14.0.1" : 9,
"2.15.0.0" : 9,
"2.15.0.1" : 9,
"latest" : 9
},
"max_python_version" : {
Expand All @@ -55,6 +59,7 @@
"2.13.0.0" : 11,
"2.14.0.1" : 11,
"2.15.0.0" : 11,
"2.15.0.1" : 11,
"latest" : 11
}
},
Expand All @@ -66,7 +71,8 @@
"2.13.0.0" : "2.13.0",
"2.14.0.1" : "2.14.0",
"2.15.0.0" : "2.15.0",
"latest" : "2.15.0"
"2.15.0.1" : "2.15.1",
"latest" : "2.15.1"
}
},
"intel_gpu_driver_list" : {
Expand Down Expand Up @@ -251,6 +257,32 @@
"1.0.15468.20-803",
"22.3.15-i803"
]
},
"2.15.0.1" : {
"ubuntu" : [
"1.3.27642.50-803~22.04",
"23.43.27642.50-803~22.04",
"1.14.0-744~22.04",
"1.0.15468.28-803~22.04",
"1.0.15468.28-803~22.04",
"22.3.15-803~22.04"
],
"rhel" : [
"1.0.15468.28-803.el8",
"1.0.15468.28-803.el8",
"22.3.15-i803.el8",
"23.43.27642.50-803.el8",
"1.14.0-791.el8",
"1.14.0-791.el8"
],
"sles" :[
"1.3.27642.50-803",
"23.43.27642.50-803",
"1.14.0-i744",
"1.0.15468.28-803",
"1.0.15468.28-803",
"22.3.15-i803"
]
}
},
"oneapi" : {
Expand Down Expand Up @@ -292,9 +324,13 @@
"dpcpp" : "2024.1.0-963",
"onemkl" : "2024.1.0-691"
},
"2.15.0.1" : {
"dpcpp" : "2024.2.1-1079",
"onemkl" : "2024.2.1-103"
},
"latest" : {
"dpcpp" : "2024.1.0-963",
"onemkl" : "2024.1.0-691"
"dpcpp" : "2024.2.1-1079",
"onemkl" : "2024.2.1-103"
}
},
"tf_requirements" : [
Expand Down

0 comments on commit 18806f1

Please sign in to comment.