torch>=1.9
tqdm
rich
nvidia-ml-py3
