diff --git a/services/ocr/Dockerfile b/services/ocr/Dockerfile index b694c61..313a38e 100644 --- a/services/ocr/Dockerfile +++ b/services/ocr/Dockerfile @@ -1,14 +1,18 @@ -FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime +FROM python:3.12-slim WORKDIR /app -# PyMuPDF/Pillow 시스템 의존성 +# PyMuPDF/Pillow 시스템 의존성 + CUDA wheel 의 런타임 로더 RUN apt-get update && apt-get install -y --no-install-recommends \ libgl1 libglib2.0-0 && \ apt-get clean && rm -rf /var/lib/apt/lists/* COPY requirements.txt . -RUN pip install --no-cache-dir -r requirements.txt +# torch / torchvision 는 PyTorch 인덱스의 cu126 wheel 고정. +# native /opt/surya-ocr/venv 검증된 조합 (torch 2.11.0+cu126, transformers 4.57.x). +RUN pip install --no-cache-dir \ + --extra-index-url https://download.pytorch.org/whl/cu126 \ + -r requirements.txt COPY server.py . diff --git a/services/ocr/requirements.txt b/services/ocr/requirements.txt index 2c78ef2..456de7a 100644 --- a/services/ocr/requirements.txt +++ b/services/ocr/requirements.txt @@ -1,3 +1,6 @@ +torch==2.11.0+cu126 +torchvision==0.26.0+cu126 +transformers==4.57.6 surya-ocr==0.17.1 pymupdf>=1.24.0,<2.0.0 fastapi>=0.110.0,<1.0.0