如何在 keras / tensorflow 挂起时重置 GPU？答案

【问题标题】：How to reset GPU on keras / tensorflow hang?如何在 keras / tensorflow 挂起时重置 GPU？
【发布时间】：2018-07-08 21:44:27
【问题描述】：

有时我必须杀死我的 python 应用程序，它使用 GPU 和 Keras 或 Tensorflow，之后我就不能再运行它们了。这可能是因为 GPU 仍然被某些东西使用。

如何在不重启机器的情况下强制释放GPU？

我尝试了以下 shell 脚本

$ cat ~/bin/nvidia-reset
#!/bin/sh

sudo rmmod nvidia_uvm

sudo rmmod nvidia_drm

sudo rmmod nvidia_modeset

sudo rmmod nvidia

sudo nvidia-smi

但通常说nvidia_uvm 很忙是无法完成工作的。

【问题讨论】：

【解决方案1】：

试试这个：

keras.backend.clear_session()

【讨论】：