sudo pip install nvidia-ml-py python3 -m pip install nvidia-ml-py
from pynvml import *
nvmlInit()
print(f"Driver Version: {nvmlSystemGetDriverVersion()}")
# Driver Version: 528.49
deviceCount = nvmlDeviceGetCount()
for i in range(deviceCount):
handle = nvmlDeviceGetHandleByIndex(i)
print(f"Device {i} : {nvmlDeviceGetName(handle)}")
# Device 0 : Tesla K40c (or:)
# Device 0 : Quadro P3200
nvmlShutdown()