diff --git a/README.md b/README.md
index 4b76650..2d50f4d 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,8 @@ The difference between our paper and ICDAR challenge is summarized [here](https:
 
 ## Getting Started
 ### Dependency
-- This work was tested with PyTorch 1.3.1, CUDA 9.0, python 3.6 and Ubuntu 16.04. <br> You may need `pip3 install torch==1.3.1`. In the paper, expriments were performed with **PyTorch 0.4.1**.
+- This work was tested with PyTorch 1.3.1, CUDA 10.1, python 3.6 and Ubuntu 16.04. <br> You may need `pip3 install torch==1.3.1`. <br>
+In the paper, expriments were performed with **PyTorch 0.4.1, CUDA 9.0**.
 - requirements : lmdb, pillow, torchvision, nltk, natsort
 ```
 pip3 install lmdb pillow torchvision nltk natsort