From e861784c86cee4f47b4d60ec48d3be9075df20b6 Mon Sep 17 00:00:00 2001 From: Hyungi Ahn Date: Thu, 23 Apr 2026 13:59:15 +0900 Subject: [PATCH] =?UTF-8?q?fix(ocr):=20align=20torch/transformers=20with?= =?UTF-8?q?=20native=20venv=20(0.17.1=20=ED=98=B8=ED=99=98=20=ED=99=95?= =?UTF-8?q?=EC=9D=B8=EB=90=9C=20=EC=A1=B0=ED=95=A9)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 이전 base image (pytorch/pytorch:2.5.1-cuda12.4) 가 surya-ocr 0.17.1 설치 시 torch 2.11.0 (PyPI CPU wheel) 로 업그레이드되지만 torchvision 0.20.1+cu124 는 유지돼 ABI 불일치 (torchvision::nms does not exist) → OCR 전체 실패. native /opt/surya-ocr/venv 에서 검증된 조합으로 복제: - python:3.12-slim base - torch 2.11.0+cu126 / torchvision 0.26.0+cu126 (PyTorch cu126 index 고정) - transformers 4.57.6 (5.x 는 surya detection.processor import 에서 실패) Co-Authored-By: Claude Opus 4.7 (1M context) --- services/ocr/Dockerfile | 10 +++++++--- services/ocr/requirements.txt | 3 +++ 2 files changed, 10 insertions(+), 3 deletions(-) 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