Zevin2023 commited on
Commit
317e5c5
1 Parent(s): 7b72a8a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -4,6 +4,7 @@ openpyxl==3.0.9
4
  Pillow==10.0.0
5
  scipy==1.11.2
6
  timm==0.5.4
7
- torch==1.11.0+cu113
8
  tqdm==4.61.2
9
  opencv-python
 
 
 
4
  Pillow==10.0.0
5
  scipy==1.11.2
6
  timm==0.5.4
 
7
  tqdm==4.61.2
8
  opencv-python
9
+ --find-links https://download.pytorch.org/whl/cu113
10
+ torch==1.11.0+gpu