torch==2.0.1
torchvision==0.15.2
numpy>=1.19.0
tqdm>=4.50.0
matplotlib>=3.7.2
pytorch-lightning>=2.0.0
tensorboard>=2.15.1
