dimwit/docker-compose.yml at main · dimwit-dev/dimwit · 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
services:
  dimwit:
    build:
      context: .
      dockerfile: Dockerfile
    image: dimwit:latest
    container_name: dimwit-dev
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - CUDA_VISIBLE_DEVICES=0
    volumes:
      # Mount the project directory
      - .:/workspace
      # Mount the SBT cache to avoid re-downloading dependencies
      - sbt-cache:/root/.sbt
      - ivy-cache:/root/.ivy2
      - coursier-cache:/root/.cache/coursier
    working_dir: /workspace
    stdin_open: true
    tty: true
    ports:
      # Jupyter notebook
      - "8888:8888"
      # Additional ports for your application
      - "5000:5000"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

volumes:
  sbt-cache:
  ivy-cache:
  coursier-cache: