Hacker-Software/ARIA-AGENT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 236 Wiki Activity

fix(f5tts): torch auf cu124 re-pinnen (Treiber 550/CUDA 12.4)

Browse Source 
f5-tts>=1.0.0 zieht als Dependency ein neueres torch mit zu neuem CUDA-Build und ueberschreibt den cu121-Pin → 'NVIDIA driver too old (found 12040)' auf Treiber 550. Nach der Installation wird dieselbe torch/torchaudio-Version als cu124-Build force-reinstalled (--no-deps), kompatibel mit 550/CUDA 12.4.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
duffyduck
2026-08-15 02:21:23 +02:00
co-authored by Claude Opus 4.8
parent ee6c4f34db
commit 7bbb75481c
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
xtts/f5tts/Dockerfile
+12
View File
@@ -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"]