torch>=1.7
tqdm
rich
nvidia-ml-py3
