服务器上自动占GPU卡的脚本

最近服务器比较挤，想train一个模型一时半会找不到空卡，要是守在屏幕前查看有没有空卡太费神了，可以让机器自己查看有沒有空卡，就写一个shell脚本，如下：

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 500 ]
        then
            echo 'GPU'$count' is avaiable'
            # 此处加上train 模型的语句
            var=1
            break
        fi
        count=$(($count+1))    
    done    
done