mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
Siddharth Kumar Sah
29a382e9e0
Add a :cuda Docker image tag that auto-detects NVIDIA GPU at runtime and falls back gracefully to CPU. Same pattern as Immich. - New gpu.py shared utility for cached CUDA detection - Background removal (rembg): pass CUDAExecutionProvider to ONNX Runtime - Upscaling (Real-ESRGAN): use CUDA device + FP16 when GPU available - OCR (PaddleOCR): enable use_gpu when CUDA detected - Dispatcher reports GPU status at startup via readiness signal - Admin health endpoint exposes GPU availability - Dockerfile uses ARG GPU=false with conditional NVIDIA CUDA base image - docker-compose.gpu.yml override for GPU users - CI/CD workflows build and publish :cuda tag (amd64 only) Three tags: :latest (CPU), :lite (no AI), :cuda (GPU with CPU fallback)
11 lines
199 B
Plaintext
11 lines
199 B
Plaintext
rembg==2.0.62
|
|
realesrgan==0.3.0
|
|
lama-cleaner==1.2.5
|
|
paddleocr==2.9.1
|
|
paddlepaddle-gpu==3.0.0
|
|
mediapipe==0.10.21
|
|
onnxruntime-gpu==1.20.1
|
|
numpy==1.26.4
|
|
Pillow==11.1.0
|
|
opencv-python-headless==4.10.0.84
|