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()