pynvml_gpu_memory_info_and_driver_version_quickstart.py

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