自动占用GPU脚本

var=0
while [ $var -eq 0 ]
do
    count=0
    for i in $(nvidia-smi --query-gpu=memory.used --format=csv,noheader,nounits)
    do
        if [ $i -lt 6000 ]
        then
            echo 'GPU'$count' is avaiable'
            sh run_mono_phoenix.sh
            var=1
            break
        fi
        count=$(($count+1))    
    done    
done