请问这个问题有知道怎么解决的吗?使用的是TensorFlow 1.11,TensorFlow-gpu 1.11,cudnn7.6.5,gcc 4.9
(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
(0)
暂无评论
亲~登录后才可以操作哦!
确定你的邮箱还未认证,请认证邮箱或绑定手机后进行当前操作
举报
×
侵犯我的权益
×
侵犯了我企业的权益
×
抄袭了我的内容
×
原文链接或出处
诽谤我
×
对根叔社区有害的内容
×
不规范转载
×
举报说明
暂无评论