torch==2.1.0
torchvision==0.9.1
opencv-python-headless==4.7.0.72
scikit-learn==1.2.2
numpy==1.24.1
kornia==0.6.12
torchsde==0.2.5
