tpu_info_quickstart_list_all_local_tpu_chips.py

1import tpu_info
2
3def main():
4    # Get information for all TPU chips on the system
5    chips = tpu_info.get_chips()
6    
7    print(f"Found {len(chips)} TPU chip(s):")
8    for chip in chips:
9        print(f"Chip ID: {chip.chip_id}")
10        print(f"  Device: {chip.device_path}")
11        print(f"  TPU Type: {chip.tpu_type}")
12        print("-" * 20)
13
14if __name__ == "__main__":
15    main()