jhj0517 commited on
Commit
3bf9562
1 Parent(s): b5e4671

Update comments in requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,5 +1,7 @@
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
 
 
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