diff --git a/setup.py b/setup.py
index e4b7c5c..e5317d7 100644
--- a/setup.py
+++ b/setup.py
@@ -96,6 +96,7 @@
     long_description=long_description,
     license='Apache 2.0',
     install_requires=[
+        'numpy==1.26.4',
         'torch',
         'torchvision',
         'Pillow>=9.0',