testintg / docker-compose.yaml
Daniton's picture
Upload 18 files
649bf38
raw
history blame
466 Bytes
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"]