torch==1.9.0
torchvision==0.7.0