diff --git a/setup.py b/setup.py
index 1864989b..fd4e44d0 100644
--- a/setup.py
+++ b/setup.py
@@ -39,8 +39,8 @@
         'numpy>=1.24.0,<2.0.0',
         'Pillow>=10.2.0,<10.3.0',
         'requests>=2.0.0,<3.0.0',
-        'torch>=2.2.0,<=2.3.0',
-        'torchvision>=0.17.0,<=0.18.0',
+        'torch>=2.2.0,<=2.4.0',
+        'torchvision>=0.17.0,<=0.19.0',
         'tqdm>=4.0.0,<5.0.0',
     ],
 )
diff --git a/tests/actions_requirements.txt b/tests/actions_requirements.txt
index b74924a5..30e699e9 100644
--- a/tests/actions_requirements.txt
+++ b/tests/actions_requirements.txt
@@ -1,7 +1,5 @@
 numpy>=1.24.0,<2.0.0
 requests>=2.0.0,<3.0.0
-torch>=2.2.0,<2.3.0
-torchvision>=0.17.0,<0.18.0
 Pillow>=10.2.0,<10.3.0
 opencv-python>=4.9.0
 scipy>=1.10.0,<2.0.0