numpy
tqdm
torch>=1.1.0
torchvision
tensorboardX
gpustat
pyyaml
addict
yapf
termcolor
tabulate
pynvml
