torch>=2.3.1
torchvision>=0.18.1
numpy<2
tqdm>=4.50.0
matplotlib>=3.7.2
pytorch-lightning==2.0.0
tensorboard>=2.15.1
