-
Notifications
You must be signed in to change notification settings - Fork 20
/
docker-compose.runtime.yml
160 lines (147 loc) · 7.78 KB
/
docker-compose.runtime.yml
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
version: "3.9"
x-base-settings: &base_service_settings
tty: true
env_file: .env
network_mode: host
build: &base_build_settings
context: .
environment: &base_environment_settings
# Colorize the terminal in the container if possible
TERM: "${TERM:-}"
deploy:
resources:
reservations:
devices:
- capabilities:
- gpu
x-main-service-settings: &main_service_settings
<<: *base_service_settings
build: &main_build_settings
<<: *base_build_settings
args: &main_build_args
DEVEL_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-devel-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-main
BUILD_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-devel-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-packages
environment: &main_environment_settings
<<: *base_environment_settings
# Use the host's X11 display
DISPLAY: "${DISPLAY:-}"
XAUTHORITY: "${XAUTHORITY:-}"
XDG_SESSION_TYPE: "${XDG_SESSION_TYPE:-}"
services:
base:
<<: *base_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
build:
<<: *base_build_settings
dockerfile: dev/dockerfiles/runtime/base.Dockerfile
args:
UID: ${UID:-1000}
AMD64_BASE: nvidia/cuda:${CUDA_VERSION:-11.6.2}-base-${LINUX_VERSION:-ubuntu20.04}
ARM64_BASE: nvidia/cuda:${CUDA_VERSION:-11.6.2}-base-${LINUX_VERSION:-ubuntu20.04}
DEVEL_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-devel-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-main
main:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-main
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/main.Dockerfile
args:
<<: *main_build_args
UCX_VERSION: ${UCX_VERSION:-1.12.1}
LINUX_VERSION: ${LINUX_VERSION:-ubuntu20.04}
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
demo:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-demo
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/demo.Dockerfile
args:
<<: *main_build_args
UCX_VERSION: ${UCX_VERSION:-1.12.1}
LINUX_VERSION: ${LINUX_VERSION:-ubuntu20.04}
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
glfw:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-glfw
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/glfw.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
cudf:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-cudf
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/cudf.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
sql:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-sql
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/sql.Dockerfile
args:
<<: *main_build_args
UCX_VERSION: ${UCX_VERSION:-1.12.1}
LINUX_VERSION: ${LINUX_VERSION:-ubuntu20.04}
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
cuml:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-cuml
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/cuml.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
cugraph:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-cugraph
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/cugraph.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
cuspatial:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-cuspatial
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/cuspatial.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-base
notebook:
<<: *main_service_settings
image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-notebook
build:
<<: *main_build_settings
dockerfile: dev/dockerfiles/runtime/notebook.Dockerfile
args:
<<: *main_build_args
FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-22.12.2}-runtime-node${NODE_VERSION:-16.15.1}-cuda${CUDA_VERSION_MAJOR:-11}-${LINUX_VERSION:-ubuntu20.04}-demo
volumes:
- "/etc/fonts:/etc/fonts:ro"
- "/tmp/.X11-unix:/tmp/.X11-unix:rw"
- "/usr/share/fonts:/usr/share/fonts:ro"
- "/usr/share/icons:/usr/share/icons:ro"
- "/etc/timezone:/etc/timezone:ro"
- "/etc/localtime:/etc/localtime:ro"
- "/run/dbus/system_bus_socket:/run/dbus/system_bus_socket"
- "${XDG_RUNTIME_DIR:-/run/user/$UID}:${XDG_RUNTIME_DIR:-/run/user/$UID}"
environment:
<<: *main_environment_settings
NVIDIA_DRIVER_CAPABILITIES: all
# Use the host's X11 display
XDG_RUNTIME_DIR: "${XDG_RUNTIME_DIR:-/run/user/$UID}"
DBUS_SESSION_BUS_ADDRESS: "${DBUS_SESSION_BUS_ADDRESS:-unix:path=/run/user/$UID/bus}"
cap_add:
- SYS_ADMIN
security_opt:
- apparmor=unconfined