torch==1.8.1
torchvision==0.9.1
tensorboard==2.4.1
