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 上分配的內存。