PuddleHub显存释放问题
收藏
PuddleHub调用图像识别模型时,识别完成后似乎并不会释放显存,最终有多少占多少导致显存溢出,大家有碰到过这种情况吗?
1、程序运行开始时候的显存占用
2、运行一次后的显存占用
3、运行两次后的显存占用
显存溢出
0
收藏
请登录后评论
进程结束后重试看看。 哇! 双卡2080!
结束进程就可以释放,但是这个是做成一个api的,总是释放不能满足预测要求呀
这个问题在PaddleHub的github上已有issue进行反馈说明:
https://github.com/PaddlePaddle/PaddleHub/issues/682
好的好的,非常感谢您