diff --git a/xtts/f5tts/Dockerfile b/xtts/f5tts/Dockerfile index 81c08ba..4bb6165 100644 --- a/xtts/f5tts/Dockerfile +++ b/xtts/f5tts/Dockerfile @@ -16,6 +16,18 @@ RUN pip3 install --no-cache-dir torch==2.3.1 torchaudio==2.3.1 \ COPY requirements.txt . RUN pip3 install --no-cache-dir -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 . CMD ["python3", "bridge.py"]