Update README.md

WJinhai · Feb 25, 2020 · 8e7aba3 · 8e7aba3
1 parent ffc3e01
commit 8e7aba3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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