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.