rrjia opened a new issue #19718: URL: https://github.com/apache/incubator-mxnet/issues/19718
Hi, I use the C++ APIs of MXNet to do some classification tasks. But I find that MXNet does not release GPU memory when calling MXPredFree(). When my program exits, all occupied GPU memory is released. So when I run my program, I can do classification just once, otherwise GPU memory is not enough. I think this is a very serious problem. I hope MXNet developers @tqchen @mli @piiswrong can look into this issue and resolve it ASAP. Thanks! The sequence of function calls is as follows: (1) MXPredCreate (2) MXPredSetInput (3) MXPredForward (4) MXPredGetOutput (5) MXPredFree after i call MXPredCreate the GPU memory will capture by mxnet, by i can not release gpu memory bu mxnet c_predict_api ---------------------------------------------------------------- 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] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
