File size: 633 Bytes
16f5171 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | 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
|