linux下kill显卡上的程序

nvidia--smi

nvidia-smi命令显示出每块显卡上运行程序的主线程PID，首先找到所想要kill掉的程序的主线程（ps -ef | grep train.py），然后kill掉主线程即可，kill掉主线程之后，再ps -ef | grep train.py，可以看到还会有很多子线程残留，以前我都是一个子线程一个子线程地kill掉，现在发现不用，kill掉主线程之后，等待一段时间，系统会自动回收由那个主线程所产生的所有子线程。