fix(f5tts): torch fest auf 2.6.0+cu124 pinnen (neuester cu124-Build)

Adaptiver Re-Pin scheiterte: f5-tts zieht torch 2.13.0, aber cu124-Wheels enden bei 2.6.0 (torch 2.7+ nur noch cu126+, was Treiber 550/CUDA12.4 nicht kann). Jetzt: torch/torchaudio 2.6.0+cu124 vor f5-tts installiert, Constraint-Datei haelt f5-tts vom Hochziehen ab. Treiber-Upgrade bleibt der strategische Fix fuer Voxtral/modernes CUDA.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
2026-08-15 02:25:12 +02:00
co-authored by Claude Opus 4.8
parent 7bbb75481c
commit 3a8202d2de
+8 -16
View File
@@ -9,24 +9,16 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
WORKDIR /app WORKDIR /app
# PyTorch CUDA-Wheels zuerst (f5-tts zieht sonst CPU-only Torch rein) # torch FEST auf cu124 (Treiber 550 = CUDA 12.4). 2.6.0 ist der NEUESTE cu124-
RUN pip3 install --no-cache-dir torch==2.3.1 torchaudio==2.3.1 \ # Build — torch 2.7+ gibt es nur noch fuer cu126+, was 550 nicht unterstuetzt
--index-url https://download.pytorch.org/whl/cu121 # ("NVIDIA driver too old, found 12040"). Der Constraint haelt f5-tts davon ab,
# torch beim Dependency-Aufloesen wieder auf eine zu neue CUDA-Version zu ziehen.
RUN pip3 install --no-cache-dir torch==2.6.0 torchaudio==2.6.0 \
--index-url https://download.pytorch.org/whl/cu124
COPY requirements.txt . COPY requirements.txt .
RUN pip3 install --no-cache-dir -r requirements.txt RUN printf 'torch==2.6.0\ntorchaudio==2.6.0\n' > /tmp/torch-constraint.txt && \
pip3 install --no-cache-dir -c /tmp/torch-constraint.txt -r requirements.txt
# f5-tts zieht als Dependency ein neueres torch (Default-Wheel, gegen ein zu neues
# CUDA gebaut) und ueberschreibt den cu121-Pin oben → "NVIDIA driver is too old
# (found version 12040)" auf Treiber 550/CUDA 12.4. Fix: DIESELBE torch-Version,
# aber als cu124-Build (kompatibel mit Treiber 550) erzwingen.
RUN set -e; \
TV="$(python3 -c 'import torch;print(torch.__version__.split(chr(43))[0])')"; \
AV="$(python3 -c 'import torchaudio;print(torchaudio.__version__.split(chr(43))[0])')"; \
echo "Re-pin torch==$TV torchaudio==$AV -> cu124"; \
pip3 install --no-cache-dir --force-reinstall --no-deps \
"torch==$TV" "torchaudio==$AV" \
--index-url https://download.pytorch.org/whl/cu124
COPY bridge.py . COPY bridge.py .