yizhangliu commited on
Commit
48103de
·
1 Parent(s): 45e015c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -15,11 +15,14 @@ setuptools
15
  supervision
16
  termcolor
17
  timm
18
- torch
19
- torchvision
20
  # torch==2.0.0
21
  # torchvision==0.15.1
22
 
 
 
 
23
  yapf
24
  numba
25
  scipy
 
15
  supervision
16
  termcolor
17
  timm
18
+ # torch
19
+ # torchvision
20
  # torch==2.0.0
21
  # torchvision==0.15.1
22
 
23
+ torch==2.1.0+cu118
24
+ torchvision==0.16.0+cu118
25
+
26
  yapf
27
  numba
28
  scipy