diff --git a/README.md b/README.md
index 91a5729..c8a24dc 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@ IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2020 <br>
 
 ## Prerequisites
 - Python 3.6
-- Pytorch 1.0 
+- Pytorch 0.4.1
 - Other required packages in `requirement.txt`