mirror of https://github.com/hpcaitech/ColossalAI
[workflow] fixed incomplete bash command (#5272)
parent
ef4f0ee854
commit
04244aaaf1
|
@ -23,6 +23,7 @@ jobs:
|
||||||
ngpu=$(nvidia-smi --query-gpu=name --format=csv,noheader | wc -l)
|
ngpu=$(nvidia-smi --query-gpu=name --format=csv,noheader | wc -l)
|
||||||
endIndex=$(($ngpu-1))
|
endIndex=$(($ngpu-1))
|
||||||
for i in $(seq 0 $endIndex);
|
for i in $(seq 0 $endIndex);
|
||||||
|
do
|
||||||
gpu_used=$(nvidia-smi -i $i --query-gpu=memory.used --format=csv,noheader,nounits)
|
gpu_used=$(nvidia-smi -i $i --query-gpu=memory.used --format=csv,noheader,nounits)
|
||||||
[ "$gpu_used" -gt "2000" ] && avai=false
|
[ "$gpu_used" -gt "2000" ] && avai=false
|
||||||
done
|
done
|
||||||
|
|
Loading…
Reference in New Issue