用法
get_tpu_system_metadata()
返回 TPU 係統的元數據。
用戶可以調用此方法來獲取 TPU 係統的一些信息,例如內核總數、TPU 工作器數量和設備。例如:
resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
tpu_system_metadata = resolver.get_tpu_system_metadata()
num_hosts = tpu_system_metadata.num_hosts
相關用法
- Python tf.distribute.cluster_resolver.TPUClusterResolver.connect用法及代碼示例
- Python tf.distribute.cluster_resolver.TPUClusterResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.TFConfigClusterResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.GCEClusterResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.SimpleClusterResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.UnionResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.SlurmClusterResolver用法及代碼示例
- Python tf.distribute.cluster_resolver.KubernetesClusterResolver用法及代碼示例
- Python tf.distribute.OneDeviceStrategy.experimental_distribute_values_from_function用法及代碼示例
- Python tf.distribute.TPUStrategy用法及代碼示例
- Python tf.distribute.experimental_set_strategy用法及代碼示例
- Python tf.distribute.experimental.MultiWorkerMirroredStrategy.gather用法及代碼示例
- Python tf.distribute.experimental.MultiWorkerMirroredStrategy用法及代碼示例
- Python tf.distribute.TPUStrategy.experimental_assign_to_logical_device用法及代碼示例
- Python tf.distribute.NcclAllReduce用法及代碼示例
- Python tf.distribute.OneDeviceStrategy.experimental_distribute_dataset用法及代碼示例
- Python tf.distribute.experimental.rpc.Server.create用法及代碼示例
- Python tf.distribute.experimental.MultiWorkerMirroredStrategy.experimental_distribute_dataset用法及代碼示例
- Python tf.distribute.OneDeviceStrategy.gather用法及代碼示例
注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.distribute.cluster_resolver.TPUClusterResolver.get_tpu_system_metadata。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。