deepxde
deepxde copied to clipboard
How can I choose a specific backend?
I have several available backends (tensorflow.compat.v1, pytorch, etc.), but how to choose a specific one? When I run my program it selects tensorflow.compat.v1 by default, but I want to use pytorch.
Use the DDE_BACKEND environment variable:
https://deepxde.readthedocs.io/en/latest/user/installation.html#working-with-different-backends
thank you for your help, I have succeeded!
import os
os.environ['DDE_BACKEND'] = 'tensorflow.compat.v1' print(os.getenv('DDE_BACKEND'))