diff --git a/setup.py b/setup.py
index 2cb266f599..aa6220ab4a 100644
--- a/setup.py
+++ b/setup.py
@@ -83,7 +83,7 @@ def package_files(prefix: str, directory: str, extension: str):
     'torchvision>=0.18.0,<0.20.2',
     'torch>=2.3.0,<2.5.2',
     'requests>=2.26.0,<3',
-    'numpy>=1.21.5,<2.2.0',
+    'numpy>=1.21.5,<2.3.0',
     'psutil>=5.8.0,<7',
     'coolname>=1.1.0,<3',
     'tabulate==0.9.0',  # for auto-generating tables