torchvision
thop
wandb
tensorboard
tqdm
py-cpuinfo
pytest