compose.base.yml 558 B

1234567891011121314151617181920212223
  1. services:
  2. app-base:
  3. build:
  4. context: .
  5. dockerfile: docker/Dockerfile
  6. args:
  7. BACKEND: ${BACKEND:-cuda} # or cpu
  8. UV_VERSION: ${UV_VERSION:-0.8.15}
  9. volumes:
  10. - ./checkpoints:/app/checkpoints
  11. - ./references:/app/references
  12. environment:
  13. COMPILE: ${COMPILE:-0}
  14. # GPU (remove this block if CPU-only):
  15. deploy:
  16. resources:
  17. reservations:
  18. devices:
  19. - driver: nvidia
  20. count: all
  21. capabilities: [gpu]
  22. tty: true
  23. stdin_open: true