marcel/familienarchiv
1
0
Fork 0
Code Issues 105 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix(ocr): install torchvision from CPU index alongside torch
Some checks failed
CI / Unit & Component Tests (push) Failing after 3s
Details
CI / Backend Unit Tests (push) Failing after 1s
Details
CI / Unit & Component Tests (pull_request) Failing after 2s
Details
CI / Backend Unit Tests (pull_request) Failing after 1s
Details

Browse Source 
torchvision installed from PyPI expects CUDA torch operator
registrations. Installing from the CPU whl index ensures torchvision
matches the CPU-only torch build. Fixes 'torchvision::nms does not
exist' RuntimeError on startup.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Marcel
2026-04-12 19:46:37 +02:00
parent 0af4749677
commit 37abc376ec
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ocr-service/Dockerfile
View File

@@ -9,8 +9,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# PyTorch CPU-only — separate layer; the whl/cpu index strips all CUDA variants (~2 GB saved) # PyTorch CPU-only — separate layer; the whl/cpu index strips all CUDA variants (~2 GB saved)
# torchvision must also come from the CPU index to match torch's operator registrations
RUN pip install --no-cache-dir \ RUN pip install --no-cache-dir \
torch==2.7.1 \ torch==2.7.1 \
torchvision==0.22.1 \
--index-url https://download.pytorch.org/whl/cpu --index-url https://download.pytorch.org/whl/cpu
COPY requirements.txt . COPY requirements.txt .

1
ocr-service/requirements.txt
View File

@@ -2,6 +2,7 @@ fastapi[standard]==0.115.6
surya-ocr==0.17.1 surya-ocr==0.17.1
kraken==7.0 kraken==7.0
torch==2.7.1 torch==2.7.1
torchvision==0.22.1
transformers>=4.56.1,<5.0.0 transformers>=4.56.1,<5.0.0
pillow>=10.2.0,<11.0.0 pillow>=10.2.0,<11.0.0
pypdfium2==4.30.0 pypdfium2==4.30.0