pynvml_gpu_info_driver_version_and_memory_usage.py

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