torch>=1.8
torchvision>=0.9
numpy
tqdm
psutil
tensorboard
packaging
pre-commit
rich
click
