torch>=2.0.1
torchvision>=0.15.2
pillow>=9.4.0
numpy>=1.26.4
matplotlib