matplotlib>=3.10.1
mpltern>=1.0.4
numpy<=2.1.2,>=2.0.0
torch>=2.6.0
torchvision>=0.21.0

[cpu]
torch==2.6.0

[cuda12]
torch==2.6.0
