中断模型训练后 GPU 显存占用未释放

ps -ef | grep train.py | grep -v grep | awk {'print $2'} | xargs -n1 kill -9

ps -ef 

显示所有正在运行的进程

|

管道

grep train.py

字符串搜索

grep -v grep

去除包含 grep 的进程行

awk {'print $2'}

只保留 pid 信息

xargs -n1 kill -9

把前面命令的输出结果作为 "kill -9" 命令的参数。