Does DeepXDE automatically use the GPU if a GPU is available?

[rtali-ia]

Hi There,

I am interested to know if DeepXDE automatically uses the GPU if a GPU is available? In case not, is there a particular setting that we need to do, in order to enable NN training on the GPU?

Thanks

[rtali-ia]

Seems like the answer is Yes. DeepXDE is automatically using the GPU