| version: "3.8" | |
| services: | |
| app: | |
| image: cogvideo | |
| build: | |
| context: . | |
| dockerfile: ./Dockerfile | |
| args: | |
| PYTHON_VERSION: ${PYTHON_VERSION:-3.9.13} | |
| environment: | |
| SAT_HOME: /work/pretrained | |
| network_mode: host | |
| tty: true | |
| volumes: | |
| - ${PWD}:/work | |
| working_dir: /work | |
| ipc: host | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| capabilities: ["gpu"] | |