docker-compose.gpu.yml

services:
  mnemo-ocr-gpu:
    build:
      context: .
      dockerfile: Dockerfile.gpu
      args:
        PADDLE_PACKAGE: paddlepaddle-gpu==3.3.0
        PADDLE_INDEX_URL: https://www.paddlepaddle.org.cn/packages/stable/cu126/
    ports:
      - "7860:7860"
    environment:
      FRONTEND_MODE: studio
      LOG_LEVEL: info
      PORT: 7860
      MNEMO_DETECTOR_MODE: colab-pipeline
      MNEMO_PADDLE_DEVICE: gpu
      MNEMO_EASYOCR_DEVICE: cpu
      MNEMO_SENSOR_OCR_ENGINE: paddle
    volumes:
      - ./dataset:/app/dataset
      - ./lib:/app/lib
      - ./input:/app/input
      - paddleocr-cache:/root/.paddleocr
      - torch-cache:/root/.cache/torch
    gpus: all
    shm_size: "8gb"

volumes:
  paddleocr-cache:
  torch-cache: