pynvml # most system have the following lib. matplotlib torch py-cpuinfo