torch>=1.9.0
matplotlib>=3.3.0
numpy>=1.19.0
seaborn>=0.11.0
pynvml>=11.0.0
tqdm>=4.60.0
