We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fd6f855 commit 7143f5eCopy full SHA for 7143f5e
1 file changed
requirements.txt
@@ -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
10
cachetools==4.2.4
11
certifi==2024.7.4
12
chardet==4.0.0
@@ -25,8 +34,5 @@ rsa==4.7.2
25
34
six==1.16.0
26
35
uritemplate==3.0.1
27
36
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
31
37
openai-whisper
32
38
platformdirs
0 commit comments