feat: re-enable GPU support for Vision Encoder after nvidia-container-toolkit installation
This commit is contained in:
@@ -225,6 +225,14 @@ services:
|
|||||||
networks:
|
networks:
|
||||||
- dagi-network
|
- dagi-network
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
# GPU support - requires nvidia-container-toolkit
|
||||||
|
deploy:
|
||||||
|
resources:
|
||||||
|
reservations:
|
||||||
|
devices:
|
||||||
|
- driver: nvidia
|
||||||
|
count: 1
|
||||||
|
capabilities: [gpu]
|
||||||
healthcheck:
|
healthcheck:
|
||||||
test: ["CMD", "curl", "-f", "http://localhost:8001/health"]
|
test: ["CMD", "curl", "-f", "http://localhost:8001/health"]
|
||||||
interval: 30s
|
interval: 30s
|
||||||
@@ -250,6 +258,37 @@ services:
|
|||||||
timeout: 10s
|
timeout: 10s
|
||||||
retries: 3
|
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:
|
volumes:
|
||||||
rag-model-cache:
|
rag-model-cache:
|
||||||
driver: local
|
driver: local
|
||||||
|
|||||||
Reference in New Issue
Block a user