torch>=1.2
tqdm
tensorboardX==1.8
ninja
numpy
psutil
packaging
