diff --git a/README.md b/README.md
index 2c0db36..5f45035 100644
--- a/README.md
+++ b/README.md
@@ -18,6 +18,8 @@ Here are some refinement results on high-resolution images.
 
 ## Quick Start
 
+Tested on PyTorch 1.0 -- though higher versions would likely work for inference as well.
+
 Check out [this folder](https://github.com/hkchengrex/CascadePSP/tree/master/segmentation-refinement). We have built a pip package that can refine an input image with two lines of code.
 
 Install with
diff --git a/docs/installation.md b/docs/installation.md
index a7d88f2..2dfdf9f 100644
--- a/docs/installation.md
+++ b/docs/installation.md
@@ -2,7 +2,7 @@
 
 ## Dependencies
 
-CascadePSP is tested on PyTorch 1.0 though higher version would likely work as well. 
+CascadePSP is tested on PyTorch 1.0 though higher version would likely work for inference as well.
 For installing PyTorch, please refer to the [Pytorch website](https://pytorch.org/).  
 
 We recommend using the anaconda distribution which should contain most of the required dependencies. Other dependencies can be installed by: