ray.util.tpu.get_num_tpu_chips_on_node#

ray.util.tpu.get_num_tpu_chips_on_node() → int[source]#

Return the number of TPU chips on the node. :returns: The total number of chips on the TPU node. Returns 0 if none are found.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.