Spaces:
Runtime error
Runtime error
jhj0517
commited on
Commit
•
3bf9562
1
Parent(s):
b5e4671
Update comments in requirements.txt
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
-
# Remove line below if you're not using
|
2 |
-
# If you're using it,
|
|
|
|
|
3 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
4 |
|
5 |
|
|
|
1 |
+
# Remove the --extra-index-url line below if you're not using Nvidia GPU.
|
2 |
+
# If you're using it, Update CUDA version to yours :
|
3 |
+
# For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
|
4 |
+
# For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
|
5 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
6 |
|
7 |
|