deepmd.tf.env
Module that sets tensorflow working environment and exports inportant constants.
Module Contents
Functions
| Limit tensorflow session to CPU or not. |
| Cast tensor to globally set TF precision. |
| Cast tensor to globally set energy precision. |
Attributes
- deepmd.tf.env.reset_default_tf_session_config(cpu_only: bool)[source]
Limit tensorflow session to CPU or not.
- Parameters:
- cpu_onlybool
If enabled, no GPU device is visible to the TensorFlow Session.