tensorboardX>=1.8
torch>=1.1.0
torchvision>=0.3.0