torch
torchvision
transforms
numpy==1.24.1
pillow