diff --git a/requirements.txt b/requirements.txt
index d3ece66ff8..3e863f05ba 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,4 +5,5 @@ scikit-learn==0.20.2
 tqdm==4.32.1
 twine==1.13.0
 numpy==1.16.4
-torch>=1.1.0
\ No newline at end of file
+torch>=1.1.0
+torchvision==0.3.0