獲取可見物理設備的列表。
用法
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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。