PINN deepXDE

修改deepXDE后端支持的深度学习框架：

• 使用环境变量：DDEBACKEND

  • 您可以使用来指定后端：DDEBACKEND=BACKEND python pde.py

  $ DDEBACKEND=tensorflow.compat.v1 python pde.py

  $ DDEBACKEND=tensorflow python pde.py

  $ DDEBACKEND=pytorch python pde.py

  $ DDEBACKEND=jax python pde.py

  • 或设置全局环境变量export DDEBACKEND=BACKEND

• 修改“~/.deepxde”下的文件：config.json

  • 该文件具有诸如{"backend": "tensorflow.compat.v1"}
  • 您还可以使用来设置默认后端python -m deepxde.backend.set_default_backend BACKEND

在第二种方法中.deepxde文件的修改需要打开隐藏文件，对于我的情况，我的anaconda安装在D盘，deepxde是创建的虚拟环境。其中.deepxde文件是在 C:\Users\13559文件下，修改其中的config.json文件中内容为相应的后端