fix: align dependency versions with FaceFusion 3.5.3 and fix GPU build

- Update onnxruntime to 1.24.1, numpy to 2.2.1, opencv-python to 4.13.0.92,
  scipy to 1.17.0, onnx to 1.20.1 to match facefusion/requirements.txt
- GPU Dockerfile: install Python 3.12 from deadsnakes PPA since CUDA base
  image ships Python 3.10 which is too old for onnxruntime-gpu 1.24.1
- Copy both requirements files in COPY step so -r reference resolves

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-02-16 13:16:21 +01:00
parent 800edc08ea
commit 4856599b57
4 changed files with 22 additions and 13 deletions

View File

@@ -7,20 +7,29 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
ffmpeg curl libgl1-mesa-glx libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
COPY requirements-cpu.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt && rm /tmp/requirements.txt
COPY requirements-cpu.txt /tmp/requirements-cpu.txt
COPY requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements-cpu.txt && rm /tmp/requirements*.txt
# ---- GPU base ----
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 AS base-gpu
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 python3-pip python3-venv \
software-properties-common \
&& add-apt-repository ppa:deadsnakes/ppa \
&& apt-get update \
&& apt-get install -y --no-install-recommends \
python3.12 python3.12-venv python3.12-dev \
ffmpeg curl libgl1-mesa-glx libglib2.0-0 \
&& ln -sf /usr/bin/python3.12 /usr/bin/python3 \
&& ln -sf /usr/bin/python3 /usr/bin/python \
&& python3 -m ensurepip --upgrade \
&& python3 -m pip install --no-cache-dir --upgrade pip \
&& rm -rf /var/lib/apt/lists/*
COPY requirements-gpu.txt /tmp/requirements.txt
RUN pip install --no-cache-dir --break-system-packages -r /tmp/requirements.txt && rm /tmp/requirements.txt
COPY requirements-gpu.txt /tmp/requirements-gpu.txt
COPY requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements-gpu.txt && rm /tmp/requirements*.txt
# ---- Final stage ----
FROM base-${VARIANT} AS final

View File

@@ -1,2 +1,2 @@
-r requirements.txt
onnxruntime==1.20.1
onnxruntime==1.24.1

View File

@@ -1,2 +1,2 @@
-r requirements.txt
onnxruntime-gpu==1.20.1
onnxruntime-gpu==1.24.1

View File

@@ -4,9 +4,9 @@ python-multipart==0.0.18
pydantic-settings==2.7.1
psutil==6.1.1
# FaceFusion dependencies (excluding gradio)
numpy==1.26.4
opencv-python==4.10.0.84
onnx==1.17.0
scipy==1.14.1
tqdm==4.67.1
# FaceFusion dependencies (matching facefusion/requirements.txt, excluding gradio)
numpy==2.2.1
opencv-python==4.13.0.92
onnx==1.20.1
scipy==1.17.0
tqdm==4.67.3