anirudh2290 opened a new issue #14686: Memory profiling enhancements URL: https://github.com/apache/incubator-mxnet/issues/14686 I have created prototype for visualizing the memory pools on the gpu. I have added a doc explaning the feature and how to use the prototype in the cwiki: https://cwiki.apache.org/confluence/display/MXNET/MXNet+Memory+Profiling+Enhancements I would need some help making this prototype ready to be PR'ed. There are more improvements that can be done as mentioned in the cwiki. Listing some of them here: 1. Support for visualizing cuDNN memory allocation and frees 2. Better visualization for CPU memory pools 3. Support for MKLDNN Memory allocation 4. Parameter server, server and worker memory visualization. Let me know if interested.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
