Skip to content

Commit 7143f5e

Browse files
Fix coda-enabled PyTorch install
1 parent fd6f855 commit 7143f5e

1 file changed

Lines changed: 9 additions & 3 deletions

File tree

requirements.txt

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,12 @@
1+
# CUDA-enabled PyTorch packages (CUDA 12.6)
2+
torch==2.3.0+cu126
3+
torchvision==0.18.0+cu126
4+
torchaudio==2.3.0+cu126
5+
6+
# Ensure the extra index for PyTorch CUDA wheels
7+
--extra-index-url https://download.pytorch.org/whl/cu126
8+
9+
# Other dependencies
110
cachetools==4.2.4
211
certifi==2024.7.4
312
chardet==4.0.0
@@ -25,8 +34,5 @@ rsa==4.7.2
2534
six==1.16.0
2635
uritemplate==3.0.1
2736
urllib3==2.2.2
28-
torch==2.3.0+cu126 --index-url https://download.pytorch.org/whl/cu126
29-
torchvision==0.18.0+cu126 --index-url https://download.pytorch.org/whl/cu126
30-
torchaudio==2.3.0+cu126 --index-url https://download.pytorch.org/whl/cu126
3137
openai-whisper
3238
platformdirs

0 commit comments

Comments
 (0)