42 lines
844 B
YAML
42 lines
844 B
YAML
|
|
services:
|
|
|
|
init-db:
|
|
build:
|
|
context: .
|
|
dockerfile: initdb.Dockerfile
|
|
container_name: infobot-init
|
|
depends_on:
|
|
db:
|
|
condition: service_healthy
|
|
volumes:
|
|
- ./scripts:/scripts:ro
|
|
- ./data/hf_cache:/root/.cache/huggingface
|
|
environment:
|
|
DB_HOST: db
|
|
DB_PORT: 5432
|
|
DB_USER: infobot
|
|
DB_PASSWORD: infobot
|
|
DB_NAME: infobot
|
|
entrypoint: ["python", "/scripts/init_and_load.py"]
|
|
devices:
|
|
- /dev/nvidia-uvm
|
|
- /dev/nvidia-uvm-tools
|
|
- /dev/nvidia-modeset
|
|
- /dev/nvidiactl
|
|
- /dev/nvidia0
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: ["gpu"]
|
|
networks: ["dumant"]
|
|
|
|
|
|
networks:
|
|
dumant:
|
|
name: dumant
|
|
external: true
|
|
|