提交 a72b5b51 编写于 作者: H hypox64

Fix some bug

上级 0d603645
......@@ -130,9 +130,10 @@ def get_gpu_use():
infos_str = infos_str.replace('|','')
infos_str = infos_str.split('\n')
print_flag = False
print(infos_str)
for line in infos_str:
line_split = line.split()
if 'No running' not in line and float(line_split[4].replace('MiB',''))>500:
if 'No running' not in line and'+-------' not in line and float(line_split[4].replace('MiB',''))>500:
task_infos = task_infos+line[2:]+'\n'
print_flag = True
if not print_flag:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册