pynvml_gpu_info_driver_version_memory_stats.py

1from pynvml import *
2
3nvmlInit()
4try:
5    print("Driver Version:", nvmlSystemGetDriverVersion())
6    deviceCount = nvmlDeviceGetCount()
7    for i in range(deviceCount):
8        handle = nvmlDeviceGetHandleByIndex(i)
9        info = nvmlDeviceGetMemoryInfo(handle)
10        print(f"Device {i}: {nvmlDeviceGetName(handle)}")
11        print(f"  Memory Total: {info.total}")
12        print(f"  Memory Free: {info.free}")
13        print(f"  Memory Used: {info.used}")
14finally:
15    nvmlShutdown()