torch
pynvml
matplotlib
numpy
colorama
torchprofile
