File size: 466 Bytes
649bf38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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"]
|