@@ -24,15 +24,8 @@ services:
24
24
no_proxy : " "
25
25
BASE_IMAGE_NAME : ${BASE_IMAGE_NAME:-ubuntu}
26
26
BASE_IMAGE_TAG : ${BASE_IMAGE_TAG:-22.04}
27
- CCL_VER : ${CCL_VER:-2021.13.1-31}
28
- DPCPP_VER : ${DPCPP_VER:-2024.2.1-1079}
29
27
GITHUB_RUN_NUMBER : ${GITHUB_RUN_NUMBER:-0}
30
- ICD_VER : ${ICD_VER:-24.22.29735.27-914~22.04}
31
- LEVEL_ZERO_DEV_VER : ${LEVEL_ZERO_DEV_VER:-1.17.6-914~22.04}
32
- LEVEL_ZERO_GPU_VER : ${LEVEL_ZERO_GPU_VER:-1.3.29735.27-914~22.04}
33
- LEVEL_ZERO_VER : ${LEVEL_ZERO_VER:-1.17.6-914~22.04}
34
28
MINIFORGE_VERSION : ${MINIFORGE_VERSION:-Linux-x86_64}
35
- MKL_VER : ${MKL_VER:-2024.2.1-103}
36
29
NO_PROXY : ' '
37
30
PACKAGE_OPTION : ${PACKAGE_OPTION:-pip}
38
31
PYTHON_VERSION : ${PYTHON_VERSION:-3.10}
@@ -41,20 +34,6 @@ services:
41
34
context : .
42
35
labels :
43
36
dependency.python : ${PYTHON_VERSION:-3.10}
44
- dependency.apt.build-essential : true
45
- dependency.apt.clinfo : true
46
- dependency.apt.git : true
47
- dependency.apt.gnupg2 : true
48
- dependency.apt.gpg-agent : true
49
- dependency.apt.intel-level-zero-gpu : ${LEVEL_ZERO_GPU_VER:-1.3.29735.27-914~22.04}
50
- dependency.apt.intel-oneapi-runtime-ccl : ${CCL_VER:-2021.13.1-31}
51
- dependency.apt.intel-oneapi-runtime-dpcpp-cpp : ${DPCPP_VER:-2024.2.1-1079}
52
- dependency.apt.intel-oneapi-runtime-mkl : ${MKL_VER:-2024.2.1-103}
53
- dependency.apt.intel-opencl-icd : ${ICD_VER:-23.43.27642.40-803~22.04}
54
- dependency.apt.level-zero : ${LEVEL_ZERO_VER:-1.17.6-914~22.04}
55
- dependency.apt.level-zero-dev : ${LEVEL_ZERO_DEV_VER:-1.17.6-914~22.04}
56
- dependency.apt.rsync : true
57
- dependency.apt.unzip : true
58
37
dependency.idp.pip : false
59
38
dependency.python.pip : requirements.txt
60
39
docs : jax
@@ -67,6 +46,7 @@ services:
67
46
bash -c "python -c 'import jax; print(\"Jax Version:\", jax.__version__)'"
68
47
depends_on :
69
48
- ${PACKAGE_OPTION:-pip}
49
+ - xpu
70
50
image : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-xla-${INTEL_XLA_VERSION:-v0.4.0}-base
71
51
pull_policy : always
72
52
jupyter :
0 commit comments