File size: 929 Bytes
6c016cc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
version: "3.8"

x-defaults: &default-gpu-settings
  environment:
    FORCE_CUDA: 0
    NVIDIA_VISIBLE_DEVICES: all # Use all GPUs, or specify GPU IDs e.g., '0,1'
  # runtime: nvidia
  deploy:
    resources:
      reservations:
        devices:
          - driver: nvidia
            count: "all"
            capabilities: [gpu]

services:
  som:
    <<: *default-gpu-settings
    build:
      context: .
      dockerfile: Dockerfile
    restart: always
    ports:
      - 8080:6092
    healthcheck:
      test: ["CMD", "curl", "--fail", "http://localhost:6092"]
      interval: 10s
      timeout: 5s
      retries: 3
      start_period: 10s

  ghcr:
    <<: *default-gpu-settings
    image: ghcr.io/feedback-box/som:main
    restart: always
    ports:
      - 8080:6092
    healthcheck:
      test: ["CMD", "curl", "--fail", "http://localhost:6092"]
      interval: 10s
      timeout: 5s
      retries: 3
      start_period: 10s