diff --git a/requirements.txt b/requirements.txt
index a25434ea9..dfbfbe187 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,3 +7,4 @@ easydict
 matplotlib
 pyyaml
 tensorboardX
+torch==0.4