torch
torchvision
numpy<2
