torch>=1.10
torchvision
numpy
