torch>=2.0.0
numpy>=1.20.0
matplotlib>=3.5.0
seaborn>=0.11.0
tqdm>=4.60.0
scipy>=1.7.0

[cuda]
torch>=2.0.0
