Shreyas Meher commited on
Commit
18e47fb
·
1 Parent(s): d386d50

Remove torch from requirements.txt to preserve GPU installs

Browse files
Files changed (2) hide show
  1. README.md +11 -1
  2. requirements.txt +4 -1
README.md CHANGED
@@ -107,7 +107,17 @@ On Windows, if you get a permission error, run PowerShell as Administrator and e
107
  Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine
108
  ```
109
 
110
- 3. Install dependencies:
 
 
 
 
 
 
 
 
 
 
111
 
112
  ```bash
113
  pip install -r requirements.txt
 
107
  Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine
108
  ```
109
 
110
+ 3. Install PyTorch:
111
+
112
+ ```bash
113
+ # CPU only (Mac, or no NVIDIA GPU):
114
+ pip install torch
115
+
116
+ # NVIDIA GPU (Windows/Linux):
117
+ pip install torch --index-url https://download.pytorch.org/whl/cu124
118
+ ```
119
+
120
+ 4. Install remaining dependencies:
121
 
122
  ```bash
123
  pip install -r requirements.txt
requirements.txt CHANGED
@@ -1,4 +1,7 @@
1
- torch
 
 
 
2
  tensorflow
3
  transformers>=4.40,<5
4
  gradio
 
1
+ # Install PyTorch separately FIRST:
2
+ # CPU only: pip install torch
3
+ # NVIDIA GPU: pip install torch --index-url https://download.pytorch.org/whl/cu124
4
+ # Then: pip install -r requirements.txt
5
  tensorflow
6
  transformers>=4.40,<5
7
  gradio