From dd9528f53c9bcf148842ab94f14a4ebd5ae223ec Mon Sep 17 00:00:00 2001 From: Siddharth Kumar Sah Date: Fri, 10 Apr 2026 10:44:47 +0800 Subject: [PATCH] fix: use PaddlePaddle GPU package index for CUDA wheels paddlepaddle-gpu==3.0.0 is not on PyPI, it is hosted on PaddlePaddle's own package index. Added -i flag pointing to the cu126 stable index for the amd64 GPU build. --- docker/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/Dockerfile b/docker/Dockerfile index 41f2fa72..32d37f87 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -101,6 +101,7 @@ RUN if [ "$TARGETARCH" = "amd64" ]; then \ /opt/venv/bin/pip install realesrgan==0.3.0 \ --extra-index-url https://download.pytorch.org/whl/cu126 && \ /opt/venv/bin/pip install paddlepaddle-gpu==3.0.0 paddleocr==2.9.1 \ + -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ \ ; else \ /opt/venv/bin/pip install "rembg[cpu]==2.0.62" && \ /opt/venv/bin/pip install realesrgan==0.3.0 && \