cudatoolkit>=10
numpy<2,>=1.20
pytorch<2,>=1.9
torchvision<1,>=0.10
