pynvml
matplotlib
torch
py-cpuinfo
fastai