[kill tensorflow] killall ipykernel processes that occupies GPU memory for a long time

kill_tensorflow.sh

#! /bin/bash

MAX_COUNT=60
COUNT=0
while [ 1 ]
do
    FREE=`nvidia-smi -q | grep Free -m 1 | grep -o '[0-9]*'`
    GPU=`nvidia-smi -q | grep Gpu -m 1 | grep -o '[0-9]*'`
    echo $COUNT/$MAX_COUNT
    if [ $FREE -lt 1000 ] && [ $GPU -lt 10 ] ; then
        let "COUNT = COUNT + 1"
	if [ $COUNT -gt $MAX_COUNT ]; then
	    echo "Killall ipykernel to free the gpu memory"
	    ps -ef | grep ipykernel | grep -v grep | cut -c 9-15 | xargs kill -9
	fi
    else
	let "COUNT = 0"
    fi
    sleep 10
done