diff --git a/README.md b/README.md
index 9dd1fc1..ced477f 100644
--- a/README.md
+++ b/README.md
@@ -18,9 +18,9 @@ PPLM allows a user to flexibly plug in one or more tiny attribute models represe
 See also our [arXiv paper](https://arxiv.org/abs/1912.02164), [blog post](https://eng.uber.com/pplm), and try it out for yourself with no setup using the [Colab notebook](https://colab.research.google.com/drive/1Ux0Z4-ruiVtJ6jUk98uk6FqfvGHCOYL3).
 
 ## Setup
-
+The project require Python3.6.
 ```bash
-pip install -r requirements.txt
+pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html
 ```
 
 ## Citation
diff --git a/requirements.txt b/requirements.txt
index b711ddd..3b09106 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,10 @@
-torch==1.7.0
+torch==1.7.0+cpu
+torchvision==0.8.1+cpu
+torchaudio===0.7.0
+typing-extensions==3.7.2
+requests==2.18.4
+packaging==16.8
+filelock==3.0.12
 nltk==3.4.5
 colorama==0.4.4
 transformers==3.4.0