torch~=1.10.2
torchvision~=0.11.3
pytest~=6.2.5