问题:在运行深度学习模型的时候,总是要nvidia-smi一下看看那块显卡比较空闲,很麻烦。
解决方法:写个bash脚本,每次运行程序的时候,选择显存剩余最大的GPU。

#!/bin/bash
n=$(nvidia-smi -q -d Memory|grep -A4 GPU|grep Free |grep -Eo "[0-9]+"|awk '{{printf"%s,",$0}}')
OLD_IFS="$IFS"
IFS=","
n_list=($n)
IFS="$OLD_IFS"

max=${n_list[0]}
max_id=0

for var in ${!n_list[@]}
do
    if [[ ${max} -le ${n_list[$var]} ]];then
    max_id=${var}
    max=${n_list[${var}]}
    echo "gpu $max_id: $max MiB"
    fi
done
echo "finally chose gpu $max_id, reamain memory $max MiB"

# run your command
#python main.py --gpu_id $max_id

相关文章:

  • 2021-09-14
  • 2021-10-07
  • 2021-05-25
  • 2021-12-06
  • 2021-12-01
  • 2021-05-01
  • 2022-12-23
  • 2021-09-02
猜你喜欢
  • 2021-09-26
  • 2021-07-09
  • 2021-10-23
  • 2021-12-05
  • 2021-06-22
  • 2021-04-19
相关资源
相似解决方案