torch>=1.2
torchvision>=0.4.0
tqdm
psutil
cpufeature
tensorboardX==1.8