| 12345678910111213141516171819202122232425 |
- services:
- app-base:
- build:
- context: .
- dockerfile: docker/Dockerfile
- args:
- BACKEND: ${BACKEND:-cuda} # or cpu
- CUDA_VER: ${CUDA_VER:-12.9.0}
- UV_EXTRA: ${UV_EXTRA:-cu129}
- UV_VERSION: ${UV_VERSION:-0.8.15}
- volumes:
- - ./checkpoints:/app/checkpoints
- - ./references:/app/references
- environment:
- COMPILE: ${COMPILE:-0}
- # GPU (remove this block if CPU-only):
- deploy:
- resources:
- reservations:
- devices:
- - driver: nvidia
- count: all
- capabilities: [gpu]
- tty: true
- stdin_open: true
|