dreamprice/docker-compose.yml at master · SharathSPhD/dreamprice · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
services:
  dreamprice:
    build:
      context: .
      dockerfile: Dockerfile
    image: dreamprice:latest
    container_name: dreamprice
    runtime: nvidia
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["0"]
              capabilities: [gpu]
    volumes:
      - .:/workspace
      - pip-cache:/root/.cache/pip
      - site-packages:/usr/local/lib/python3.12/dist-packages
    ports:
      - "8000:8000"
    env_file:
      - .env
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - WANDB_PROJECT=dreamprice
      - PYTHONPATH=/workspace/src
    mem_limit: 100g
    memswap_limit: 115g
    ipc: host
    ulimits:
      memlock:
        soft: -1
        hard: -1
      stack:
        soft: 67108864
        hard: 67108864
    working_dir: /workspace
    stdin_open: true
    tty: true
    command: /bin/bash

volumes:
  pip-cache:
  site-packages: