Spaces:
Running on Zero
Running on Zero
Update requirements.txt
Browse files- requirements.txt +6 -0
requirements.txt
CHANGED
|
@@ -1,3 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
torch==2.4.1
|
| 2 |
torchaudio==2.4.1
|
| 3 |
torchvision==0.19.1
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
|
| 3 |
+
--extra-index-url https://pypi.nvidia.com
|
| 4 |
+
--trusted-host download.pytorch.org
|
| 5 |
+
--trusted-host aiinfra.pkgs.visualstudio.com
|
| 6 |
+
--trusted-host pypi.nvidia.com
|
| 7 |
torch==2.4.1
|
| 8 |
torchaudio==2.4.1
|
| 9 |
torchvision==0.19.1
|