diff --git a/requirements.txt b/requirements.txt
index 4427411..a348081 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,5 +12,5 @@ h5py
 imgaug
 IPython[all]
 pandas>=0.23.1
+torch==0.4.1
 torchvision>=0.2.1
-torch==0.4.1
\ No newline at end of file