Use torch<2
Signed-off-by: Edward Yang <94523015+edoyango@users.noreply.github.com>pull/13506/head
parent
a9f7979c83
commit
e32038aee6
|
@ -4,7 +4,7 @@
|
|||
# Base ------------------------------------------------------------------------
|
||||
gitpython>=3.1.30
|
||||
matplotlib>=3.3
|
||||
numpy>=1.18.5
|
||||
numpy>=1.18.5<2
|
||||
opencv-python>=4.1.1
|
||||
Pillow>=7.1.2,<10
|
||||
psutil # system resources
|
||||
|
@ -12,8 +12,8 @@ PyYAML>=5.3.1
|
|||
requests>=2.23.0
|
||||
scipy>=1.4.1
|
||||
thop>=0.1.1 # FLOPs computation
|
||||
torch>=1.7.0 # see https://pytorch.org/get-started/locally (recommended)
|
||||
torchvision>=0.8.1
|
||||
torch>=1.7.0<2 # see https://pytorch.org/get-started/locally (recommended)
|
||||
torchvision>=0.8.1,<0.15
|
||||
tqdm>=4.64.0
|
||||
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
|
||||
|
||||
|
|
Loading…
Reference in New Issue