Skip to content

Commit

Permalink
[Docker] update for driver version (#2719)
Browse files Browse the repository at this point in the history
  • Loading branch information
@zufangzhu
zufangzhu authored Aug 8, 2024
1 parent 569aef2 commit c45e839
Show file tree
Hide file tree
Showing 5 changed files with 173 additions and 117 deletions.
4 changes: 2 additions & 2 deletions docker/README.md
View file
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
View file
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-xpu.Dockerfile
View file
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
142 changes: 91 additions & 51 deletions tools/python/config.json
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@

{
"latest_release": "2.15.0.1",
"os_list" : [
"ubuntu",
"rhel",
Expand All @@ -14,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 @@ -24,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 @@ -34,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 @@ -45,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 @@ -54,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 @@ -65,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 @@ -173,7 +180,7 @@
"22.3.5-i601"
]
},
"2.13.0" : {
"2.13.0.0" : {
"ubuntu" : [
"1.3.26241.33-647~22.04",
"23.17.26241.33-647~22.04",
Expand All @@ -199,7 +206,7 @@
"22.3.5-i647"
]
},
"2.14.0" : {
"2.14.0.1" : {
"ubuntu" : [
"1.3.26918.50-736~22.04",
"23.30.26918.50-736~22.04",
Expand All @@ -225,7 +232,7 @@
"22.3.10-i712"
]
},
"2.15.0" : {
"2.15.0.0" : {
"ubuntu" : [
"1.3.27642.40-803~22.04",
"23.43.27642.40-803~22.04",
Expand All @@ -250,73 +257,106 @@
"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" : {
"libsycl.so" :"Intel(R) OneAPI DPC++/C++ Compiler",
"libmkl_sycl.so" : "Intel(R) OneAPI Math Kernel Library",
"libccl.so" : "Intel(R) OneAPI Collective Communications Library"
"dpcpp" : "Intel(R) OneAPI DPC++/C++ Compiler",
"onemkl" : "Intel(R) OneAPI Math Kernel Library"
},
"oneapi_lib" : [
"libsycl.so",
"libmkl_sycl.so",
"libccl.so"
],
"oneapi_version" : {
"1.0.0" : [
"2022.2.0-8734",
"2022.2.0-8748"
],
"1.1.0" : [
"2023.0.0-25370",
"2023.0.0-25398",
"2021.8.0-25371"
"oneapi_lib" : {
"dpcpp" : [
"libsycl.so"
],
"1.2.0" : [
"2023.1.0-46305",
"2023.1.0-46342",
"2021.9.0-43543"
],
"2.13.0" : [
"2023.2.0-49495",
"2023.2.0-49495",
"2021.10.0-49084"
],
"2.14.0" : [
"2024.0.0-49819",
"2024.0.0-49819",
"2021.11.0-49156"
],
"2.15.0" : [
"2024.1.0-963",
"2024.1.0-691",
"2021.12.0-309"
"onemkl" : [
"libmkl_sycl_blas.so",
"libmkl_sycl_lapack.so",
"libmkl_sycl_dft.so"
]
},
"oneapi_version" : {
"1.0.0" : {
"dpcpp" : "2022.2.0-8734",
"onemkl" : "2022.2.0-8748"
},
"1.1.0" : {
"dpcpp" : "2023.0.0-25370",
"onemkl" : "2023.0.0-25398"
},
"1.2.0" : {
"dpcpp" : "2023.1.0-46305",
"onemkl" : "2023.1.0-46342"
},
"2.13.0.0" : {
"dpcpp" : "2023.2.0-49495",
"onemkl" : "2023.2.0-49495"
},
"2.14.0.1" : {
"dpcpp" : "2024.0.0-49819",
"onemkl" : "2024.0.0-49819"
},
"2.15.0.0" : {
"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.2.1-1079",
"onemkl" : "2024.2.1-103"
}
},
"tf_requirements" : [
"absl-py",
"absl",
"astunparse",
"flatbuffers",
"gast",
"google-pasta",
"grpcio",
"pasta",
"grpc",
"h5py",
"Keras|keras-nightly",
"libclang",
"keras",
"clang",
"numpy",
"opt-einsum",
"opt_einsum",
"packaging",
"protobuf",
"google.protobuf",
"six",
"termcolor",
"typing_extensions",
"tb-nightly|tensorboard",
"tensorflow-estimator|tf-estimator-nightly",
"tensorflow-io-gcs-filesystem",
"tensorboard",
"tensorflow_estimator",
"tensorflow_io_gcs_filesystem",
"wrapt"
],
"itex_requirements" : [
"wheel",
"requests"
]
}
}
Loading

0 comments on commit c45e839

Please sign in to comment.