From e32038aee64fd324379a9aa11987e2557357b681 Mon Sep 17 00:00:00 2001
From: Edward Yang <94523015+edoyango@users.noreply.github.com>
Date: Mon, 3 Feb 2025 09:57:03 +1100
Subject: [PATCH] Use torch<2

Signed-off-by: Edward Yang <94523015+edoyango@users.noreply.github.com>
---
 requirements.txt | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index f19db27a0..f884e656e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -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