felipecode / article-memory-log Public

forked from quentinf00/article-memory-log

Notifications You must be signed in to change notification settings
Fork 0
Star 0

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

0 stars 6 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
src		src
README.md		README.md

Repository files navigation

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

About

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

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%