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:
+8
-16
@@ -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 .
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user