Python tf.config.set_visible_devices用法及代码示例

设置可见设备列表。

用法

tf.config.set_visible_devices(
    devices, device_type=None
)

参数

devices 要显示的PhysicalDevice 列表
device_type (可选)仅配置匹配此设备类型的设备。例如 "CPU" 或 "GPU"。其他设备将保持不变。

抛出

ValueError 如果参数验证失败。
RuntimeError 运行时已经初始化。

指定哪些PhysicalDevice 对象对运行时可见。 TensorFlow 只会在可见的物理设备上分配内存和放置操作，否则不会在它们上创建 LogicalDevice。默认情况下，所有发现的设备都标记为可见。

以下示例演示了禁用机器上的第一个 GPU。

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable first GPU
  tf.config.set_visible_devices(physical_devices[1:], 'GPU')
  logical_devices = tf.config.list_logical_devices('GPU')
  # Logical device was not created for first GPU
  assert len(logical_devices) == len(physical_devices) - 1
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.config.set_visible_devices。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。