Update docker-compose.yml

VIDA-NYU · Dec 21, 2023 · 8a5ad34 · 8a5ad34
1 parent 7a2e2b5
commit 8a5ad34
Showing 1 changed file with 64 additions and 64 deletions.
diff --git a/registry/docker-compose.yml b/registry/docker-compose.yml
@@ -86,79 +86,79 @@ 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
 
   raw_recorder:
     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