torch>=1.6
tqdm
psutil
