update README.md

README.md

@@ -1,12 +1,12 @@
 # aTEAM
 A pyTorch Extension for Applied Mathematics
 
-This version is compatible with pytorch (0.4.0). You can create a conda environment for pytorch0.4.0:
+This version is compatible with pytorch (1.0.1). You can create a conda environment for pytorch1.0:
 ```
-conda create -n torch0.4 python=3 jupyter
-source activate torch0.4
-conda install pytorch=0.4
-# or conda install pytorch-cpu=0.4
+conda create -n torch1.0 python=3 jupyter
+source activate torch1.0
+conda install pytorch=1.0 cudatoolkit=9.0 -c pytorch
+# or conda install pytorch-cpu=1.0 -c pytorch
 ```
 
 ## Some code maybe useful to you:
@@ -20,17 +20,25 @@ For more usages pls refer to aTEAM/test/*.py
 
 # PDE-Net
 
-Initially, aTEAM is written for this paper:
+aTEAM is a basic library for PDE-Net & PDE-Net 2.0[(source code)](https://github.com/ZichaoLong/PDE-Net):
 
-[PDE-Net: Learning PDEs from Data](https://arxiv.org/abs/1710.09668)[(ICML 2018)](https://icml.cc/Conferences/2018)[(source code)](https://github.com/ZichaoLong/PDE-Net)<br />
-[Long Zichao](http://zlong.me/), [Lu Yiping](http://about.2prime.cn/), [Ma Xianzhong](https://www.researchgate.net/profile/Xianzhong_Ma), [Dong Bin](http://bicmr.pku.edu.cn/~dongbin)
+- [PDE-Net: Learning PDEs from Data](https://arxiv.org/abs/1710.09668)[(ICML 2018)](https://icml.cc/Conferences/2018)<br />
+[Long Zichao](http://zlong.me/), [Lu Yiping](https://web.stanford.edu/~yplu/), [Ma Xianzhong](https://www.researchgate.net/profile/Xianzhong_Ma), [Dong Bin](http://bicmr.pku.edu.cn/~dongbin)
+- [PDE-Net 2.0: Learning PDEs from Data with A Numeric-Symbolic Hybrid Deep Network](https://arxiv.org/abs/1812.04426)<br />
+[Long Zichao](http://zlong.me/), [Lu Yiping](https://web.stanford.edu/~yplu/), [Dong Bin](http://bicmr.pku.edu.cn/~dongbin)
 
-If you find this code useful in your research then please cite
+If you find this code useful for your research then please cite
 ```
 @inproceedings{long2017pde,
     title={PDE-Net: Learning PDEs from Data},
     author={Long, Zichao and Lu, Yiping and Ma, Xianzhong and Dong, Bin},
     booktitle={Proceedings of the 35th International Conference on Machine Learning (ICML 2018)},
     year={2018}
 }
+@article{long2018pde,
+    title={PDE-Net 2.0: Learning PDEs from Data with A Numeric-Symbolic Hybrid Deep Network},
+    author={Long, Zichao and Lu, Yiping and Dong, Bin},
+    journal={arXiv preprint arXiv:1812.04426},
+    year={2018}
+}
 ```