Spaces:
Runtime error
Runtime error
jhj0517
commited on
Commit
•
cb60fd1
1
Parent(s):
237a214
Give space
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
-
# Remove
|
2 |
-
# If you're using it,
|
3 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
|
|
|
|
4 |
torch
|
5 |
torchvision
|
6 |
pytest
|
|
|
1 |
+
# Remove line below if you're not using nvidia GPU.
|
2 |
+
# If you're using it, update the CUDA version at the end to yours. For example /cu121 , /cu118 etc..
|
3 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
4 |
+
|
5 |
+
|
6 |
torch
|
7 |
torchvision
|
8 |
pytest
|