設置是否應為 PhysicalDevice 啟用內存增長。
用法
tf.config.experimental.set_memory_growth(
device, enable
)參數
-
devicePhysicalDevice進行配置 -
enable(布爾值)是否啟用或禁用內存增長
拋出
-
ValueError指定的PhysicalDevice無效。 -
RuntimeError運行時已經初始化。
如果為 PhysicalDevice 啟用內存增長，則運行時初始化將不會分配設備上的所有內存。無法在配置了虛擬設備的 PhysicalDevice 上配置內存增長。
例如：
physical_devices = tf.config.list_physical_devices('GPU')
try:
tf.config.experimental.set_memory_growth(physical_devices[0], True)
except:
# Invalid device or cannot modify virtual devices once initialized.
pass相關用法
- Python tf.config.experimental.get_memory_usage用法及代碼示例
- Python tf.config.experimental.get_memory_info用法及代碼示例
- Python tf.config.experimental.enable_tensor_float_32_execution用法及代碼示例
- Python tf.config.experimental.enable_op_determinism用法及代碼示例
- Python tf.config.experimental.get_device_details用法及代碼示例
- Python tf.config.experimental.ClusterDeviceFilters用法及代碼示例
- Python tf.config.experimental.reset_memory_stats用法及代碼示例
- Python tf.config.experimental.get_memory_growth用法及代碼示例
- Python tf.config.experimental_connect_to_cluster用法及代碼示例
- Python tf.config.experimental_connect_to_host用法及代碼示例
- Python tf.config.list_logical_devices用法及代碼示例
- Python tf.config.list_physical_devices用法及代碼示例
- Python tf.config.get_logical_device_configuration用法及代碼示例
- Python tf.config.run_functions_eagerly用法及代碼示例
- Python tf.config.set_visible_devices用法及代碼示例
- Python tf.config.set_logical_device_configuration用法及代碼示例
- Python tf.config.get_visible_devices用法及代碼示例
- Python tf.concat用法及代碼示例
- Python tf.convert_to_tensor用法及代碼示例
- Python tf.constant_initializer.from_config用法及代碼示例
注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.config.experimental.set_memory_growth。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。