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.
This commit is contained in:
Siddharth Kumar Sah
2026-04-10 10:44:47 +08:00
parent caf65bc469
commit dd9528f53c
+1
View File
@@ -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 && \