通过Apache mod_wsgi django调用keras模型时，如何释放占用的GPU内存? [英] How to release the occupied GPU memory when calling keras model by Apache mod_wsgi django?

问题描述

我的服务器配置如下:

1. Apache 2.4.23.
2. Mod_wsgi 4.5.9

1. apache 2.4.23.
2. Mod_wsgi 4.5.9

通过使用Django框架和apache服务器，我们称为Keras深度学习模型. 成功调用模型后，该模型始终在GPU内存中运行，这导致除非关闭apache服务器，否则无法释放GPU内存.

By using the Django framework and apache server, we call the Keras deep learning model. And after the successful calling of the model, the model has been always running in the GPU memory, which causes the GPU memory can not be released except by shutting down the apache server.

因此，在通过Apache + Mod_wsgi + Django调用Keras模型时，有什么方法可以控制GPU内存的释放?

谢谢！

运行时内存占用量屏幕截图

推荐答案

from keras import backend as K
K.clear_session()

这将清除当前会话(图形)，因此应从GPU中删除过时的模型.如果它不起作用，则可能需要删除模型"并再次重新加载.

This will clear the current session (Graph) and so the stale model should be removed from GPU. If it didn't work, you might need to 'del model' and reload it again.

这篇关于通过Apache mod_wsgi django调用keras模型时，如何释放占用的GPU内存?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！