numpy
setuptools
torchvision