Use torch<2

Signed-off-by: Edward Yang <94523015+edoyango@users.noreply.github.com>
pull/13506/head
Edward Yang 2025-02-03 09:57:03 +11:00 committed by GitHub
parent a9f7979c83
commit e32038aee6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions

View File

@ -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