PriorBLIP/docker-compose.yml at master · yuanmouya-prog/PriorBLIP · 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
version: '3.8'

services:
  priorblip-train:
    build:
      context: .
      dockerfile: Dockerfile
    image: priorblip:latest
    container_name: priorblip-training
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - CUDA_VISIBLE_DEVICES=0,1,2,3,4,5
      - PYTHONPATH=/workspace/src:$PYTHONPATH
      - HF_ENDPOINT=https://hf-mirror.com
    volumes:
      # 挂载数据目录
      - /data/yy:/data/yy:ro
      # 挂载源代码（开发模式）
      - ./src:/workspace/src
      - ./scripts:/workspace/scripts
      # 挂载训练日志
      - ./logs:/workspace/logs
      # 挂载数据集（如果需要）
      # - /data/yy/RSICD:/data/RSICD:ro
      # - /data/yy/RSITMD:/data/RSITMD:ro
    working_dir: /workspace
    stdin_open: true
    tty: true
    command: /bin/bash
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  # TensorBoard服务
  tensorboard:
    image: tensorflow/tensorflow:latest
    container_name: priorblip-tensorboard
    ports:
      - "6006:6006"
    volumes:
      - ./logs:/workspace/logs
    command: tensorboard --logdir=/workspace/logs --host=0.0.0.0 --port=6006
    profiles:
      - monitoring