torch
torchvision
numpy
packaging
