diff --git a/requirements.txt b/requirements.txt
index 55f6a68..b669270 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+numpy==1.26.4
 torch
 torchvision
 Pillow>=9.0
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',