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