-
Notifications
You must be signed in to change notification settings - Fork 18
/
Copy pathdocker-compose.yaml
67 lines (66 loc) · 2.74 KB
/
docker-compose.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Copyright (c) 2024 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
include:
- path:
- ../python/docker-compose.yaml
services:
jax-base:
build:
args:
http_proxy: ${http_proxy}
https_proxy: ${https_proxy}
no_proxy: ""
BASE_IMAGE_NAME: ${BASE_IMAGE_NAME:-ubuntu}
BASE_IMAGE_TAG: ${BASE_IMAGE_TAG:-22.04}
GITHUB_RUN_NUMBER: ${GITHUB_RUN_NUMBER:-0}
MINIFORGE_VERSION: ${MINIFORGE_VERSION:-Linux-x86_64}
NO_PROXY: ''
PACKAGE_OPTION: ${PACKAGE_OPTION:-pip}
PYTHON_VERSION: ${PYTHON_VERSION:-3.10}
REGISTRY: ${REGISTRY}
REPO: ${REPO}
context: .
labels:
dependency.python: ${PYTHON_VERSION:-3.10}
dependency.idp.pip: false
dependency.python.pip: requirements.txt
docs: jax
org.opencontainers.base.name: "intel/python:3.10-core"
org.opencontainers.image.name: "intel/intel-optimized-xla"
org.opencontainers.image.title: "Intel® Optimized XLA Base Image"
org.opencontainers.image.version: ${INTEL_XLA_VERSION:-v0.4.0}-${PACKAGE_OPTION:-pip}-base
target: jax-base
command: >
bash -c "python -c 'import jax; print(\"Jax Version:\", jax.__version__)'"
depends_on:
- ${PACKAGE_OPTION:-pip}
- xpu
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
pull_policy: always
jupyter:
build:
labels:
dependency.python.pip: jupyter-requirements.txt
org.opencontainers.base.name: "intel/intel-optimized-xla:${INTEL_XLA_VERSION:-v0.4.0}-base"
org.opencontainers.image.title: "Intel® Optimized XLA Jupyter Base Image"
org.opencontainers.image.version: ${INTEL_XLA_VERSION:-v0.4.0}-jupyter
target: jupyter
command: >
bash -c "python -m jupyter --version"
environment:
http_proxy: ${http_proxy}
https_proxy: ${https_proxy}
extends: jax-base
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}-jupyter
network_mode: host