roman-bachmann commited on
Commit
2c57d1b
1 Parent(s): b2425c2

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -24,8 +24,8 @@ scikit-image
24
  imutils
25
 
26
  -f https://download.pytorch.org/whl/torch_stable.html
27
- torch==1.10.0
28
- torchvision==0.11.1
29
 
30
  timm==0.4.12
31
  einops==0.3.2
 
24
  imutils
25
 
26
  -f https://download.pytorch.org/whl/torch_stable.html
27
+ torch==1.10.0+cu111
28
+ torchvision==0.11.1+cu111
29
 
30
  timm==0.4.12
31
  einops==0.3.2