mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
7 lines
168 B
7 lines
168 B
2 years ago
|
avai=true
|
||
|
|
for i in $(seq 0 7);
|
||
|
|
do
|
||
|
|
gpu_used=$(nvidia-smi -i $i --query-gpu=memory.used --format=csv,noheader,nounits)
|
||
|
|
[ "$gpu_used" -le "10000" ] && avai=false
|
||
|
|
done
|