node/docker-compose.devel.yml at main · rapidsai/node · GitHub

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
x-base-settings: &base_service_settings
  tty: true
  env_file:
    - .env
  network_mode: host
  build: &base_build_settings
    context: .
    x-bake:
      platforms:
        - linux/arm64
      output: type=docker
  environment: &base_environment_settings
    # Colorize the terminal in the container if possible
    TERM: "${TERM:-linux}"
    NVIDIA_DISABLE_REQUIRE: true
    NVIDIA_DRIVER_CAPABILITIES: all
  deploy:
    resources:
      reservations:
        devices:
          - capabilities:
            - gpu

x-main-service-settings: &main_service_settings
  <<: *base_service_settings
  environment: &main_environment_settings
    <<: *base_environment_settings
    # Use the host's X11 display
    DISPLAY: "${DISPLAY:-}"
    XAUTHORITY: "${XAUTHORITY:-}"
    XDG_SESSION_TYPE: "${XDG_SESSION_TYPE:-}"
    XDG_RUNTIME_DIR: "${XDG_RUNTIME_DIR:-/run/user/$UID}"
    LIBCUDF_KERNEL_CACHE_PATH: "/opt/rapids/node/.cache/jit"
    RAPIDS_VERSION: "${npm_package_config_rapids_version:-26.04.00}"
    DBUS_SESSION_BUS_ADDRESS: "${DBUS_SESSION_BUS_ADDRESS:-unix:path=/run/user/$UID/bus}"
  cap_add:
    - SYS_ADMIN
    - SYS_PTRACE
  security_opt:
    - apparmor=unconfined

services:

  main: &main_service
    <<: *main_service_settings
    image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-26.4.0}-devel-node${NODE_VERSION:-24.14.1}-cuda${CUDA_VERSION_MAJOR:-12}-${LINUX_VERSION:-ubuntu24.04}-main
    build:
      <<: *base_build_settings
      dockerfile: dev/dockerfiles/devel/main.Dockerfile
      args:
        UCX_VERSION: ${UCX_VERSION:-1.20.0}
        NODE_VERSION: ${NODE_VERSION:-24.14.1}
        LINUX_VERSION: ${LINUX_VERSION:-ubuntu24.04}
        AMD64_BASE: nvidia/cuda:${CUDA_VERSION:-12.9.1}-devel-${LINUX_VERSION:-ubuntu24.04}
        ARM64_BASE: nvidia/cuda:${CUDA_VERSION:-12.9.1}-devel-${LINUX_VERSION:-ubuntu24.04}
    environment:
      <<: *main_environment_settings
      SSH_AUTH_SOCK: "/ssh-agent"
      GCC_COLORS: "${GCC_COLORS:-error=01;31:warning=01;35:note=01;36:caret=01;32:locus=01:quote=01}"
    ulimits:
      nofile:
        soft: 500000
        hard: 500000
    volumes:
      - ".:/opt/rapids/node:rw"
      - "/etc/fonts:/etc/fonts:ro"
      - "/etc/timezone:/etc/timezone"
      - "/etc/localtime:/etc/localtime"
      - "/tmp/.X11-unix:/tmp/.X11-unix:rw"
      - "/usr/share/fonts:/usr/share/fonts:ro"
      - "/usr/share/icons:/usr/share/icons:ro"
      - "/run/dbus/system_bus_socket:/run/dbus/system_bus_socket"
      - "${SSH_AUTH_SOCK:-/run/user/$UID/keyring/ssh}:/ssh-agent"
      - "${XDG_RUNTIME_DIR:-/run/user/$UID}:${XDG_RUNTIME_DIR:-/run/user/$UID}"

  packages:
    <<: *base_service_settings
    image: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-26.4.0}-devel-node${NODE_VERSION:-24.14.1}-cuda${CUDA_VERSION_MAJOR:-12}-${LINUX_VERSION:-ubuntu24.04}-packages
    build:
      <<: *base_build_settings
      dockerfile: dev/dockerfiles/devel/package.Dockerfile
      ssh:
        - default
      secrets:
        - aws_creds
        - sccache_config
      args:
        CUDAARCHS: "${CUDAARCHS:-RAPIDS}"
        PARALLEL_LEVEL: "${PARALLEL_LEVEL:-}"
        NVCC_APPEND_FLAGS: "${NVCC_APPEND_FLAGS:---threads=100}"
        SCCACHE_REGION: "${SCCACHE_REGION:-us-east-2}"
        SCCACHE_BUCKET: "${SCCACHE_BUCKET:-rapids-sccache-devs}"
        RAPIDS_VERSION: "${npm_package_config_rapids_version:-26.04.00}"
        FROM_IMAGE: ${REPOSITORY:-ghcr.io/rapidsai/node}:${npm_package_version:-26.4.0}-devel-node${NODE_VERSION:-24.14.1}-cuda${CUDA_VERSION_MAJOR:-12}-${LINUX_VERSION:-ubuntu24.04}-main
    ulimits:
      nofile:
        soft: 500000
        hard: 500000

secrets:
  aws_creds:
    file: ../.aws/credentials
  sccache_config:
    file: ../.config/sccache/config