python 3.6.13
matplotlib 3.3.4
numpy 1.19.5
torch 1.7.0
CuDNN 8.1.0
Cuda 10.1
[1] Raissi, Maziar, Paris Perdikaris, and George E. Karniadakis. "Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations." Journal of Computational Physics 378 (2019): 686-707.