diff --git a/registry/docker-compose.yml b/registry/docker-compose.yml index 89ab029..d4722aa 100644 --- a/registry/docker-compose.yml +++ b/registry/docker-compose.yml @@ -86,70 +86,71 @@ services: - "traefik.http.routers.tim-dash.entrypoints=web" - "traefik.http.services.tim-dash.loadbalancer.server.port=80" - # argus: - # image: ghcr.io/vida-nyu/argus2:main - # container_name: argus - # restart: unless-stopped - # ports: - # - "3000:80" - # labels: - # - "traefik.enable=true" - # - "traefik.http.routers.tim-dash.rule=Host(`argus.${DOMAIN:-localhost}`)" - # - "traefik.http.routers.tim-dash.entrypoints=web" - # - "traefik.http.services.tim-dash.loadbalancer.server.port=80" - - # # perception & reasoning - - # perception: - # image: ghcr.io/vida-nyu/ptg-server-ml-perception:main - # container_name: perception - # restart: unless-stopped - # volumes: - # - /dev/shm:/dev/shm - # - ${VOLUMES:-./data}/models:/src/app/models - # - ${VOLUMES:-./data}/models/cache:/root/.cache - # - ${VOLUMES:-./data}/models/torch_iocache:/root/.torch/iopath_cache - # environment: - # PYTHONUNBUFFERED: "1" - # # devices: - # # - /dev/nvidia1:/dev/nvidia0 - # # - /dev/nvidia0:/dev/nvidia1 - # # deploy: - # # resources: - # # reservations: - # # devices: - # # - driver: nvidia - # # count: 2 - # # capabilities: [gpu] + argus: + image: ghcr.io/vida-nyu/argus2:main + container_name: argus + restart: unless-stopped + ports: + - "3000:80" + labels: + - "traefik.enable=true" + - "traefik.http.routers.tim-dash.rule=Host(`argus.${DOMAIN:-localhost}`)" + - "traefik.http.routers.tim-dash.entrypoints=web" + - "traefik.http.services.tim-dash.loadbalancer.server.port=80" + + # perception & reasoning + + perception: + image: ghcr.io/vida-nyu/ptg-server-ml-perception:main + container_name: perception + restart: unless-stopped + volumes: + - /dev/shm:/dev/shm + - ${VOLUMES:-./data}/models:/src/app/models + - ${VOLUMES:-./data}/models/cache:/root/.cache + - ${VOLUMES:-./data}/models/torch_iocache:/root/.torch/iopath_cache + environment: + PYTHONUNBUFFERED: "1" + devices: + - /dev/nvidia0:/dev/nvidia0 + #- /dev/nvidia1:/dev/nvidia1 + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu] - # reasoning: - # image: ghcr.io/vida-nyu/ptg-server-ml-reasoning:main - # container_name: reasoning - # restart: unless-stopped - # volumes: - # - ${VOLUMES:-./data}/models/reasoning:/src/app/models - # - ${VOLUMES:-./data}/models/reasoning/nltk:/usr/share/nltk_data - # - ${VOLUMES:-./data}/models/reasoning/spacy:/opt/conda/lib/python3.8/site-packages/en_core_web_lg - # # devices: - # # - /dev/nvidia0:/dev/nvidia0 - # # deploy: - # # resources: - # # reservations: - # # devices: - # # - driver: nvidia - # # count: 1 - # # capabilities: [gpu] - - # memory: - # image: ghcr.io/vida-nyu/ptg-server-ml-3d-memory:main - # container_name: memory - # restart: unless-stopped + reasoning: + image: ghcr.io/vida-nyu/ptg-server-ml-reasoning:main + container_name: reasoning + restart: unless-stopped + volumes: + - ${VOLUMES:-./data}/models/reasoning:/src/app/models + - ${VOLUMES:-./data}/models/reasoning/nltk:/usr/share/nltk_data + - ${VOLUMES:-./data}/models/reasoning/spacy:/opt/conda/lib/python3.8/site-packages/en_core_web_lg + # devices: + # - /dev/nvidia0:/dev/nvidia0 + # deploy: + # resources: + # reservations: + # devices: + # - driver: nvidia + # count: 1 + # capabilities: [gpu] + + memory: + image: ghcr.io/vida-nyu/ptg-server-ml-3d-memory:main + container_name: memory + command: main.py mem run + restart: unless-stopped - # 3d-memory-sync: - # image: ghcr.io/vida-nyu/ptg-server-ml-3d-memory:main - # container_name: 3d-memory-sync - # command: main.py sync run - # restart: unless-stopped + 3d-memory-sync: + image: ghcr.io/vida-nyu/ptg-server-ml-3d-memory:main + container_name: 3d-memory-sync + command: main.py sync run + restart: unless-stopped # recorders @@ -157,8 +158,7 @@ services: image: ghcr.io/vida-nyu/ptg-server-ml-record:main container_name: raw_recorder restart: unless-stopped - # command: python -m ptgprocess.processors.record raw run --continuous - command: sleep 100000 + command: python -m ptgprocess.processors.record raw run --continuous volumes: - ${VOLUMES:-./data}/recordings/raw:/src/app/recordings