felipecode/article-memory-log

Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**

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