gpustat_query_all_gpus_and_print_status.py

1import gpustat
2
3# Query the current status of all GPUs
4stats = gpustat.GPUStatCollection.new_query()
5
6# Iterate through each GPU and print its information
7for gpu in stats:
8    print(gpu)
9
10# Alternatively, convert the entire collection to a dictionary or JSON
11# print(stats.jsonify())