script for grabbing gpu

run.sh

while true
do
    m=`nvidia-smi --format=csv --query-gpu=memory.used 2>&1 | awk '{if(NR>1 && NR<=2)print $1}'`
    if [ $m -lt 2000 ]
    then
        echo executing
        exec python tools/train.py --gpus 0,1 configs/res18.yaml --name res18_base
    else
        echo memory $m
        sleep 1
    fi
done