feat: re-enable GPU support for Vision Encoder after nvidia-container-toolkit installation
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user