Skip to content

Commit fd6f855

Browse files
Upgrade torch to CUDA support versions
1 parent b96f209 commit fd6f855

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

requirements.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,5 +25,8 @@ rsa==4.7.2
2525
six==1.16.0
2626
uritemplate==3.0.1
2727
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
2831
openai-whisper
2932
platformdirs

0 commit comments

Comments
 (0)