Memory profiling in pytorch Code to reproduce the results of Deep Learning CUDA Memory Usage and Pytorch optimization tricks Requirements You need a CUDA enabled GPU python 3.6+ pytorch Installation instructions apex Installation instructions matplotlib pandas Run python src/article_implem.py