GitHub

tf.config.experimental.get_memory_usage

Get the current memory usage, in bytes, for the chosen device. (deprecated)

tf.config.experimental.get_memory_usage(
    device
)

This function is deprecated in favor of tf.config.experimental.get_memory_info. Calling this function is equivalent to calling tf.config.experimental.get_memory_info()['current'].

See https://www.tensorflow.org/api_docs/python/tf/device for specifying device strings.

For example:

gpu_devices = tf.config.list_physical_devices('GPU')
if gpu_devices:
  tf.config.experimental.get_memory_usage('GPU:0')

Does not work for CPU.

For GPUs, TensorFlow will allocate all the memory by default, unless changed with tf.config.experimental.set_memory_growth. This function only returns the memory that TensorFlow is actually using, not the memory that TensorFlow has allocated on the GPU.

device Device string to get the bytes in use for, e.g. "GPU:0"

Total memory usage in bytes.

ValueError Non-existent or CPU device specified.