Skip to content

Commit b279006

Browse files
fix version pinned torch
1 parent 7143f5e commit b279006

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# 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
2+
torch==2.3.0
3+
torchvision==0.18.0
4+
torchaudio==2.3.0
55

66
# Ensure the extra index for PyTorch CUDA wheels
77
--extra-index-url https://download.pytorch.org/whl/cu126

0 commit comments

Comments
 (0)