（Linux）显存被占用-显示没有进程-nohup

作者：不正经 | 2024-05-26 08:27:40

踩

显存被占用-显示没有进程

在深度学习（基于Pytorch）任务 nohup 后台训练时，可以用nvidia-smi命令查看进程的GPU运行状态，如图

nvidia-smi
1

在这里插入图片描述

当kill掉进程后，发现进程没有了，但是显存依然被占用。这种情况多半是代码中 num_works 的设置大于1导致的。nvidia-smi显示的PID是父进程，那么在kill掉此进程后，子进程仍然在工作。关键问题在于如何找到子进程并终止。

解决办法：

ps -aux
1

在这里插入图片描述

kill $(pgrep -f "python XXX.py --XXX")
1

为了防止出现此种情况，每次想kill掉进程的时候不要直接 kill，要同时kill掉当前进程与子进程（PID代表进程号）：

kill $(pgrep -P PID)
1

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/不正经/article/detail/625965