获取可见物理设备的列表。
用法
tf.config.get_visible_devices(
device_type=None
)参数
-
device_type(可选字符串)仅包括与此设备类型匹配的设备。例如 "CPU" 或 "GPU"。
返回
-
可见
PhysicalDevice的列表
返回当前标记为对运行时可见的 PhysicalDevice 的列表。一旦运行时初始化，可见设备将至少有一个与之关联的LogicalDevice。
以下示例验证所有可见的 GPU 已被禁用：
physical_devices = tf.config.list_physical_devices('GPU')
try:
# Disable all GPUS
tf.config.set_visible_devices([], 'GPU')
visible_devices = tf.config.get_visible_devices()
for device in visible_devices:
assert device.device_type != 'GPU'
except:
# Invalid device or cannot modify virtual devices once initialized.
pass相关用法
- Python tf.config.get_logical_device_configuration用法及代码示例
- Python tf.config.list_logical_devices用法及代码示例
- Python tf.config.experimental.get_memory_usage用法及代码示例
- Python tf.config.list_physical_devices用法及代码示例
- Python tf.config.experimental.get_memory_info用法及代码示例
- Python tf.config.run_functions_eagerly用法及代码示例
- Python tf.config.experimental.enable_tensor_float_32_execution用法及代码示例
- Python tf.config.experimental_connect_to_cluster用法及代码示例
- Python tf.config.experimental.set_memory_growth用法及代码示例
- Python tf.config.experimental_connect_to_host用法及代码示例
- Python tf.config.set_visible_devices用法及代码示例
- Python tf.config.set_logical_device_configuration用法及代码示例
- 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.concat用法及代码示例
- Python tf.convert_to_tensor用法及代码示例
- Python tf.constant_initializer.from_config用法及代码示例
注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.config.get_visible_devices。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。