How to inspect the detail memory usage of mxnet


#1

Is it possible to inspect the detail GPU memory usage of mxnet? I am working on a project which adopt several deep learning framework(eg. tensorflow and pytorch), but the GPU memory usage is increasing, which is unbearable, and I don’t know which DL frame causes that.

I am wonder if there is a API that i can use it to get the GPU memory usage.

Thanks.


#2

You can use the profiler to get an accurate view of memory usage. Profiler usage is explained in this tutorial.