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"]