#torch>=1.2
#torchvision>=0.4.0
tqdm
tensorboardX==1.8
ninja
numpy
psutil