diff --git a/requirements.txt b/requirements.txt
index 390381f..f60a02d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ scipy==0.18.1
 h5py==2.6.0
 tensorboardX==0.8
 # for tensorboard web server
-tensorflow==1.2.0
\ No newline at end of file
+tensorflow==1.2.0
+torch