torch>=1.13.0
torchvision>=0.14.1
numpy
matplotlib