feat: re-enable GPU support for Vision Encoder after nvidia-container-toolkit installation

This commit is contained in:
Apple
2025-11-17 06:45:54 -08:00
parent 1fc2f7cfee
commit a945b42ba0

View File

@@ -225,6 +225,14 @@ services:
networks:
- dagi-network
restart: unless-stopped
# GPU support - requires nvidia-container-toolkit
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:8001/health"]
interval: 30s
@@ -250,6 +258,37 @@ services:
timeout: 10s
retries: 3
# Node Registry Service - Central registry for DAGI network nodes
node-registry:
build:
context: ./services/node-registry
dockerfile: Dockerfile
container_name: dagi-node-registry
ports:
- "9205:9205" # Internal network only
environment:
- NODE_REGISTRY_DB_HOST=postgres
- NODE_REGISTRY_DB_PORT=5432
- NODE_REGISTRY_DB_NAME=node_registry
- NODE_REGISTRY_DB_USER=node_registry_user
- NODE_REGISTRY_DB_PASSWORD=${NODE_REGISTRY_DB_PASSWORD:-CHANGE_ME_STRONG_PASSWORD}
- NODE_REGISTRY_HTTP_PORT=9205
- NODE_REGISTRY_ENV=production
- NODE_REGISTRY_LOG_LEVEL=info
volumes:
- ./logs:/app/logs
depends_on:
- postgres
networks:
- dagi-network
restart: unless-stopped
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:9205/health"]
interval: 30s
timeout: 10s
retries: 3
start_period: 10s
volumes:
rag-model-cache:
driver: local