Skip to content

Commit c45e839

Browse files
zufangzhuzufangzhu
authored
[Docker] update for driver version (#2719)
1 parent 569aef2 commit c45e839

File tree

5 files changed

+173
-117
lines changed

5 files changed

+173
-117
lines changed

‎docker/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@ wget <download link from https://pypi.org/project/intel-optimization-for-horovod
3131
### I. Customize Build Script
3232
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.
3333

34-
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.
34+
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.
3535

3636
```bash
3737
IMAGE_NAME=intel-extension-for-tensorflow:cpu-ubuntu
3838
docker build --build-arg UBUNTU_VERSION=22.04 \
3939
--build-arg PYTHON=python3.10 \
40-
--build-arg TF_VER=2.15 \
40+
--build-arg TF_VER=2.15.1 \
4141
--build-arg WHEELS=*.whl \
4242
-t $IMAGE_NAME \
4343
-f itex-cpu.Dockerfile .

‎docker/build.sh

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -21,23 +21,23 @@ if [ $IMAGE_TYPE == "xpu" -o $IMAGE_TYPE == "gpu" ]
2121
then
2222
IMAGE_NAME=intel-extension-for-tensorflow:$IMAGE_TYPE
2323
docker build --no-cache --build-arg UBUNTU_VERSION=22.04 \
24-
--build-arg ICD_VER=23.43.27642.40-803~22.04 \
25-
--build-arg LEVEL_ZERO_GPU_VER=1.3.27642.40-803~22.04 \
24+
--build-arg ICD_VER=23.43.27642.50-803~22.04 \
25+
--build-arg LEVEL_ZERO_GPU_VER=1.3.27642.50-803~22.04 \
2626
--build-arg LEVEL_ZERO_VER=1.14.0-744~22.04 \
2727
--build-arg LEVEL_ZERO_DEV_VER=1.14.0-744~22.04 \
28-
--build-arg DPCPP_VER=2024.1.0-963 \
29-
--build-arg MKL_VER=2024.1.0-691 \
30-
--build-arg CCL_VER=2021.12.0-309 \
28+
--build-arg DPCPP_VER=2024.2.1-1079 \
29+
--build-arg MKL_VER=2024.2.1-103 \
30+
--build-arg CCL_VER=2021.13.1-31 \
3131
--build-arg PYTHON=python3.10 \
32-
--build-arg TF_VER=2.15 \
32+
--build-arg TF_VER=2.15.1 \
3333
--build-arg WHEELS=*.whl \
3434
-t $IMAGE_NAME \
3535
-f itex-xpu.Dockerfile .
3636
else
3737
IMAGE_NAME=intel-extension-for-tensorflow:$IMAGE_TYPE
3838
docker build --no-cache --build-arg UBUNTU_VERSION=22.04 \
3939
--build-arg PYTHON=python3.10 \
40-
--build-arg TF_VER=2.15 \
40+
--build-arg TF_VER=2.15.1 \
4141
--build-arg WHEELS=*.whl \
4242
-t $IMAGE_NAME \
4343
-f itex-cpu.Dockerfile .

‎docker/itex-xpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ RUN pip --no-cache-dir install --upgrade \
9696
pip \
9797
setuptools
9898

99-
ARG TF_VER="2.15"
99+
ARG TF_VER="2.15.1"
100100

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

‎tools/python/config.json

Lines changed: 91 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11

22
{
3+
"latest_release": "2.15.0.1",
34
"os_list" : [
45
"ubuntu",
56
"rhel",
@@ -14,6 +15,7 @@
1415
"2.13.0.0" : "20.04 22.04",
1516
"2.14.0.1" : "22.04",
1617
"2.15.0.0" : "22.04",
18+
"2.15.0.1" : "22.04",
1719
"latest" : "22.04"
1820
},
1921

@@ -24,6 +26,7 @@
2426
"2.13.0.0" : "8.6",
2527
"2.14.0.1" : "8.6 8.8 9.0 9.2",
2628
"2.15.0.0" : "8.6 8.8 8.9 9.0 9.2 9.3",
29+
"2.15.0.1" : "8.6 8.8 8.9 9.0 9.2 9.3",
2730
"latest" : "8.6 8.8 8.9 9.0 9.2 9.3"
2831
},
2932

@@ -34,6 +37,7 @@
3437
"2.13.0.0" : "15.3 15.4",
3538
"2.14.0.1" : "15.4 15.5",
3639
"2.15.0.0" : "15.4 15.5",
40+
"2.15.0.1" : "15.4 15.5",
3741
"latest" : "15.4 15.5"
3842
}
3943
},
@@ -45,6 +49,7 @@
4549
"2.13.0.0" : 8,
4650
"2.14.0.1" : 9,
4751
"2.15.0.0" : 9,
52+
"2.15.0.1" : 9,
4853
"latest" : 9
4954
},
5055
"max_python_version" : {
@@ -54,6 +59,7 @@
5459
"2.13.0.0" : 11,
5560
"2.14.0.1" : 11,
5661
"2.15.0.0" : 11,
62+
"2.15.0.1" : 11,
5763
"latest" : 11
5864
}
5965
},
@@ -65,7 +71,8 @@
6571
"2.13.0.0" : "2.13.0",
6672
"2.14.0.1" : "2.14.0",
6773
"2.15.0.0" : "2.15.0",
68-
"latest" : "2.15.0"
74+
"2.15.0.1" : "2.15.1",
75+
"latest" : "2.15.1"
6976
}
7077
},
7178
"intel_gpu_driver_list" : {
@@ -173,7 +180,7 @@
173180
"22.3.5-i601"
174181
]
175182
},
176-
"2.13.0" : {
183+
"2.13.0.0" : {
177184
"ubuntu" : [
178185
"1.3.26241.33-647~22.04",
179186
"23.17.26241.33-647~22.04",
@@ -199,7 +206,7 @@
199206
"22.3.5-i647"
200207
]
201208
},
202-
"2.14.0" : {
209+
"2.14.0.1" : {
203210
"ubuntu" : [
204211
"1.3.26918.50-736~22.04",
205212
"23.30.26918.50-736~22.04",
@@ -225,7 +232,7 @@
225232
"22.3.10-i712"
226233
]
227234
},
228-
"2.15.0" : {
235+
"2.15.0.0" : {
229236
"ubuntu" : [
230237
"1.3.27642.40-803~22.04",
231238
"23.43.27642.40-803~22.04",
@@ -250,73 +257,106 @@
250257
"1.0.15468.20-803",
251258
"22.3.15-i803"
252259
]
260+
},
261+
"2.15.0.1" : {
262+
"ubuntu" : [
263+
"1.3.27642.50-803~22.04",
264+
"23.43.27642.50-803~22.04",
265+
"1.14.0-744~22.04",
266+
"1.0.15468.28-803~22.04",
267+
"1.0.15468.28-803~22.04",
268+
"22.3.15-803~22.04"
269+
],
270+
"rhel" : [
271+
"1.0.15468.28-803.el8",
272+
"1.0.15468.28-803.el8",
273+
"22.3.15-i803.el8",
274+
"23.43.27642.50-803.el8",
275+
"1.14.0-791.el8",
276+
"1.14.0-791.el8"
277+
],
278+
"sles" :[
279+
"1.3.27642.50-803",
280+
"23.43.27642.50-803",
281+
"1.14.0-i744",
282+
"1.0.15468.28-803",
283+
"1.0.15468.28-803",
284+
"22.3.15-i803"
285+
]
253286
}
254287
},
255288
"oneapi" : {
256-
"libsycl.so" :"Intel(R) OneAPI DPC++/C++ Compiler",
257-
"libmkl_sycl.so" : "Intel(R) OneAPI Math Kernel Library",
258-
"libccl.so" : "Intel(R) OneAPI Collective Communications Library"
289+
"dpcpp" : "Intel(R) OneAPI DPC++/C++ Compiler",
290+
"onemkl" : "Intel(R) OneAPI Math Kernel Library"
259291
},
260-
"oneapi_lib" : [
261-
"libsycl.so",
262-
"libmkl_sycl.so",
263-
"libccl.so"
264-
],
265-
"oneapi_version" : {
266-
"1.0.0" : [
267-
"2022.2.0-8734",
268-
"2022.2.0-8748"
269-
],
270-
"1.1.0" : [
271-
"2023.0.0-25370",
272-
"2023.0.0-25398",
273-
"2021.8.0-25371"
292+
"oneapi_lib" : {
293+
"dpcpp" : [
294+
"libsycl.so"
274295
],
275-
"1.2.0" : [
276-
"2023.1.0-46305",
277-
"2023.1.0-46342",
278-
"2021.9.0-43543"
279-
],
280-
"2.13.0" : [
281-
"2023.2.0-49495",
282-
"2023.2.0-49495",
283-
"2021.10.0-49084"
284-
],
285-
"2.14.0" : [
286-
"2024.0.0-49819",
287-
"2024.0.0-49819",
288-
"2021.11.0-49156"
289-
],
290-
"2.15.0" : [
291-
"2024.1.0-963",
292-
"2024.1.0-691",
293-
"2021.12.0-309"
296+
"onemkl" : [
297+
"libmkl_sycl_blas.so",
298+
"libmkl_sycl_lapack.so",
299+
"libmkl_sycl_dft.so"
294300
]
295301
},
302+
"oneapi_version" : {
303+
"1.0.0" : {
304+
"dpcpp" : "2022.2.0-8734",
305+
"onemkl" : "2022.2.0-8748"
306+
},
307+
"1.1.0" : {
308+
"dpcpp" : "2023.0.0-25370",
309+
"onemkl" : "2023.0.0-25398"
310+
},
311+
"1.2.0" : {
312+
"dpcpp" : "2023.1.0-46305",
313+
"onemkl" : "2023.1.0-46342"
314+
},
315+
"2.13.0.0" : {
316+
"dpcpp" : "2023.2.0-49495",
317+
"onemkl" : "2023.2.0-49495"
318+
},
319+
"2.14.0.1" : {
320+
"dpcpp" : "2024.0.0-49819",
321+
"onemkl" : "2024.0.0-49819"
322+
},
323+
"2.15.0.0" : {
324+
"dpcpp" : "2024.1.0-963",
325+
"onemkl" : "2024.1.0-691"
326+
},
327+
"2.15.0.1" : {
328+
"dpcpp" : "2024.2.1-1079",
329+
"onemkl" : "2024.2.1-103"
330+
},
331+
"latest" : {
332+
"dpcpp" : "2024.2.1-1079",
333+
"onemkl" : "2024.2.1-103"
334+
}
335+
},
296336
"tf_requirements" : [
297-
"absl-py",
337+
"absl",
298338
"astunparse",
299339
"flatbuffers",
300340
"gast",
301-
"google-pasta",
302-
"grpcio",
341+
"pasta",
342+
"grpc",
303343
"h5py",
304-
"Keras|keras-nightly",
305-
"libclang",
344+
"keras",
345+
"clang",
306346
"numpy",
307-
"opt-einsum",
347+
"opt_einsum",
308348
"packaging",
309-
"protobuf",
349+
"google.protobuf",
310350
"six",
311351
"termcolor",
312352
"typing_extensions",
313-
"tb-nightly|tensorboard",
314-
"tensorflow-estimator|tf-estimator-nightly",
315-
"tensorflow-io-gcs-filesystem",
353+
"tensorboard",
354+
"tensorflow_estimator",
355+
"tensorflow_io_gcs_filesystem",
316356
"wrapt"
317357
],
318358
"itex_requirements" : [
319359
"wheel",
320360
"requests"
321361
]
322-
}
362+
}

0 commit comments

Comments
 (0)