torch>=1.7.0
torchvision
better_exchook
