PROTEA/docker-compose.prod.yml at main · frapercan/PROTEA · 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
# Production overrides: pull pre-built images from ghcr.io instead of building locally.
# Use with: docker compose -f docker-compose.yml -f docker-compose.prod.yml up -d
#
# The worker-embeddings-batch service gets GPU access via the NVIDIA container runtime.

services:
  migrate:
    image: ghcr.io/frapercan/protea:latest

  api:
    image: ghcr.io/frapercan/protea:latest

  worker-jobs:
    image: ghcr.io/frapercan/protea:latest

  worker-embeddings:
    image: ghcr.io/frapercan/protea:latest

  worker-embeddings-batch:
    image: ghcr.io/frapercan/protea:latest
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
        limits:
          memory: 8G

  worker-embeddings-write:
    image: ghcr.io/frapercan/protea:latest

  worker-predictions-batch:
    image: ghcr.io/frapercan/protea:latest

  worker-predictions-write:
    image: ghcr.io/frapercan/protea:latest

  worker-reaper:
    image: ghcr.io/frapercan/protea:latest

  frontend:
    image: ghcr.io/frapercan/protea-frontend:latest