torch>=1.7
tqdm
nvidia-ml-py3
