pynvml
matplotlib
torch
py-cpuinfo