torch>=2.1.2
torchvision>=0.17.1

[test]
