设置是否应为 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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。