Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge


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

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 for specifying device strings.

For example:

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

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.