torch>=1.11.0
numpy
torchvision>=0.12.0
kornia==0.6.2
matplotlib
pytest
