diff --git a/requirements.txt b/requirements.txt
index 2b92917..554b013 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 numpy
-torch>=1.1
-torchvision>=0.4
+torch==2.3.1
+torchvision==0.18.1
 scikit-learn
 yacs
 tqdm
diff --git a/setup.py b/setup.py
index 5adbb27..b4a15a9 100644
--- a/setup.py
+++ b/setup.py
@@ -192,7 +192,7 @@ if __name__ == '__main__':
         ],
         license='Apache License 2.0',
         install_requires=[
-            'numpy', 'torch>=1.2', 'torchvision>=0.4', 'scikit-learn', 'yacs', 'tqdm',
+            'numpy', 'torch==2.3.1', 'torchvision==0.18.1', 'scikit-learn', 'yacs', 'tqdm',
         ],
         packages=find_packages(),
         zip_safe=False)