| version: "3.8" | |
| services: | |
| api: | |
| build: . | |
| ports: | |
| - "8000:8000" | |
| volumes: | |
| - ./configs:/app/configs | |
| - ./data:/app/data | |
| - ./checkpoints:/app/checkpoints | |
| environment: | |
| - CUDA_VISIBLE_DEVICES=0 | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: 1 | |
| capabilities: [gpu] | |
| restart: unless-stopped | |