diff --git a/setup.py b/setup.py
index e046f7e4..e63ad005 100644
--- a/setup.py
+++ b/setup.py
@@ -16,6 +16,7 @@
     packages=setuptools.find_packages(),
     install_requires=[
         'torch>=1.13.1',
+        'numpy<2',
         'torchvision>=0.13',
         'scikit-learn',
         'json5',