Fine-tuning/docker-compose.yml

version: "3.8"

services:
  postgres:
    image: postgres:16-alpine
    container_name: finetune-postgres
    restart: unless-stopped
    environment:
      POSTGRES_DB: finetuning
      POSTGRES_USER: finetune
      POSTGRES_PASSWORD: finetune123
    volumes:
      - pgdata:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    networks:
      - finetune-net

  backend:
    build:
      context: ./backend
      dockerfile: Dockerfile
    container_name: finetune-backend
    restart: unless-stopped
    ports:
      - "8010:8010"
    volumes:
      # 持久化数据和模型
      - ./backend/data:/root/Fine-tuning/backend/data
    env_file:
      - ./backend/.env.docker
    environment:
      - BACKEND_HOST=0.0.0.0
      - BACKEND_PORT=8010
      - DATABASE_URL=postgresql+asyncpg://finetune:finetune123@postgres:5432/finetuning
      # 沐曦 maca 环境变量
      - MACA_PATH=/opt/maca
      - LD_LIBRARY_PATH=/opt/maca/lib:/opt/maca/mxgpu_llvm/lib:/opt/maca/ompi/lib
      - MACA_CLANG_PATH=/opt/maca/mxgpu_llvm/bin
    depends_on:
      - postgres
    devices:
      - /dev/mxcd:/dev/mxcd
    privileged: true
    shm_size: "2gb"
    networks:
      - finetune-net

  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile
      args:
        VITE_API_BASE_URL: /api/v1
        VITE_WS_BASE_URL: /ws
    container_name: finetune-frontend
    restart: unless-stopped
    ports:
      - "3000:80"
    depends_on:
      - backend
    networks:
      - finetune-net

networks:
  finetune-net:
    driver: bridge

volumes:
  pgdata: