gpu_run.sh 2 python cnn.py

gpu_run.sh

#!/bin/bash
used=`nvidia-smi | sed 's/\s\s*/ /g' |cut -d " " -f1,2,3|grep "| [0-9]"|grep Tesla -v|cut -d" " -f2`
allgpus=`nvidia-smi | sed 's/\s\s*/ /g' |cut -d " " -f1,2,3|grep "| [0-9]"|grep Tesla|cut -d" " -f2`
#comm -12 <(echo -e $used| sort -u) <(echo -e $allgpus | sort -u)
ava=`bash -c "comm -3 <(grep -oP '\d+' <<<\"$allgpus\" | sort) <(grep -oP '\d+' <<<\"$used\" | sort) |   tr -d '\t' |   paste -sd','"`
reqed=`echo $ava|cut -d"," -f1-$1`
#echo $reqed,
cmds=`echo $@|cut -d" " -f2-`
#echo $cmds
CUDA_VISIBLE_DEVICES=$reqed $cmds