harlanhong commited on
Commit
d8320b4
1 Parent(s): 17b549f
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -28,8 +28,8 @@ sklearn==0.0
28
  tensorboard==1.15.0
29
  threadpoolctl==3.1.0
30
  tifffile==2021.11.2
31
- torch
32
- torchvision
33
  tqdm==4.64.0
34
  typing_extensions==4.2.0
35
  Werkzeug==2.1.1
 
28
  tensorboard==1.15.0
29
  threadpoolctl==3.1.0
30
  tifffile==2021.11.2
31
+ torch==1.10.1+cu111
32
+ torchvision==0.11.2+cu111
33
  tqdm==4.64.0
34
  typing_extensions==4.2.0
35
  Werkzeug==2.1.1