torch>=2.0.0
matplotlib>=3.5.0