有些网络由于驱动或者版本等问题,python会占用显存完了也不退出,只能手动清空显存,一个个杀进程太麻烦,用脚本方便
# kill python processes
ps -ef | grep -v grep | grep python | grep -v root | awk '{print $2}' | xargs kill -9
nvidia-smi | grep -v grep |grep python | awk '{print $5}' | xargs kill -9
https://github.com/EthanZhangCN/milq/blob/Ethan/scripts/bash/killpython.sh