nlp-template/compose.yml at main · iamtatsuki05/nlp-template · 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
version: "3.2"
services:
  python-cpu:
    tty: true
    stdin_open: true
    user: root
    working_dir: /app
    build:
      context: .
      dockerfile: docker/cpu/Dockerfile
      target: dev
      # secrets:
      #   - github_token
      args:
        progress: plain
    volumes:
      - type: bind
        source: ./
        target: /app
    ports:
      - "8888:8888"
    command:
      uv run jupyter lab
        --ip=0.0.0.0
        --allow-root
        --NotebookApp.token=''
        --port=8888
    environment:
        PYTHONPATH: "/app/src"
        PYTHONUNBUFFERED: 1
  python-gpu:
    tty: true
    stdin_open: true
    user: root
    working_dir: /app
    build:
      context: .
      dockerfile: docker/gpu/Dockerfile
      target: dev
      # secrets:
      #   - github_token
      args:
        progress: plain
    volumes:
      - type: bind
        source: ./
        target: /app
    ports:
      - "6006:6006"
      - "8888:8888"
    command:
      uv run jupyter lab
        --ip=0.0.0.0
        --allow-root
        --NotebookApp.token=''
        --port=8888
    shm_size: '16gb'
    runtime: nvidia
    environment:
        NVIDIA_VISIBLE_DEVICES: all
        NVIDIA_DRIVER_CAPABILITIES: all
        PYTHONPATH: "/app/src"
        PYTHONUNBUFFERED: 1
# secrets:
#   github_token:
#     file: ${HOME}/.git-credentials