首页 新闻 会员 周边 捐助

请问这个问题有知道怎么解决的吗?使用的是TensorFlow 1.11,TensorFlow-gpu 1.11,cudnn7.6.5,gcc 4.9

0
[待解决问题]

请问这个问题有知道怎么解决的吗?使用的是TensorFlow 1.11,TensorFlow-gpu 1.11,cudnn7.6.5,gcc 4.9

ss淡淡的主页 ss淡淡 | 菜鸟二级 | 园豆:208
提问于:2022-09-14 10:20
< >
分享
所有回答(1)
0

!/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

人工智能技术与咨询 | 园豆:228 (菜鸟二级) | 2022-09-14 10:57
清除回答草稿
   您需要登录以后才能回答,未注册用户请先注册