torch
torchvision
numpy==1.24.4