diff --git a/requirements.txt b/requirements.txt
index 52846b8..1100495 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@
 Cython
 numpy==1.17
 opencv-python
-torch==1.4
+torch>=1.4
 matplotlib
 pillow
 tensorboard