python - 如何简单地杀死 python-tensorflow 进程并释放内存？

Question

每当我运行一个使用 tensorflow 的 python 脚本并出于某种原因决定在它完成之前将其杀死时，就会出现 ctrl-c 不起作用的问题。我会使用 ctrl-z 但它不会释放 gpu 内存，所以当我尝试重新运行脚本时没有剩余内存。在 Linux 中有解决方案吗？

Answer 1

我总是从脚本启动 tensorflow 程序。例如:

python tf_run.py 1> ./log 2> ./err &

然后使用top/htop来监控你的程序状态。如果您的机器上还有许多其他进程，请仅排在最前面的 python 进程。

top -p $(pgrep -d',' python)

最后，当你想终止进度时，

ps aux | grep tf_run.py | awk '{print $2}' | xargs kill -9

当您有多个 tensorflow 进程时，此命令行非常有用。