torch>=1.9
matplotlib>=3.4
