fix: Use flexible PyTorch versions and fix Docker build order

- Don't pin exact torch/torchaudio/torchvision versions (use >=2.1.0)
- Install CUDA PyTorch first in Docker before other requirements
- Upgrade pip before installations to avoid compatibility issues
- Let pip resolve latest compatible versions from cu118 index
This commit is contained in:
Your Name
2026-01-05 10:50:20 -05:00
parent c5cc7c2969
commit 78e9df31e6
2 changed files with 14 additions and 12 deletions

View File

@ -8,9 +8,10 @@ humanize>=4.6.0
# PyTorch ecosystem - updated for SpeechBrain 1.0 compatibility
# torchaudio >= 2.1.0 is REQUIRED for diarization to work properly
torch==2.1.0
torchaudio==2.1.0
torchvision==0.16.0
# NOTE: For Docker GPU builds, these are installed separately from cu118 index
torch>=2.1.0
torchaudio>=2.1.0
torchvision>=0.16.0
# Transformers ecosystem - compatible versions
transformers==4.35.0