Python tf.config.experimental.get_memory_usage用法及代码示例

获取所选设备的当前内存使用情况(以字节为单位)。 (已弃用)

用法

tf.config.experimental.get_memory_usage(
    device
)

警告：此函数已弃用。它将在未来的版本中删除。更新说明：改用 tf.config.experimental.get_memory_info(device)['current']。

此函数已弃用，取而代之的是 tf.config.experimental.get_memory_info 。调用此函数等效于调用 tf.config.experimental.get_memory_info()['current'] 。

参看https://www.tensorflow.org/api_docs/python/tf/device用于指定设备字符串。

例如：

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

不适用于 CPU。

对于 GPU，TensorFlow 将默认分配所有内存，除非更改为 tf.config.experimental.set_memory_growth 。该函数只返回 TensorFlow 实际使用的内存，而不是 TensorFlow 在 GPU 上分配的内存。