Skip to content

GitLab

Commit 602e1842 authored by huchen's avatar huchen
Browse files

Merge branch 'xuan_dev' into 'develop' 

DTK-203

See merge request dcutoolkit/deeplearing/NNI!3
parents fefee175 30ac210e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/nni_trial_tool/gpu.py
View file @ 602e1842
...@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi): ...@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi):
output["gpuInfos"] = [] output["gpuInfos"] = []
for gpuIndex, gpu in enumerate(gpuList): for gpuIndex, gpu in enumerate(gpuList):
if gpu == 'system': if gpu == 'system':
break continue
gpuInfo = {} gpuInfo = {}
gpuInfo['index'] = gpuIndex gpuInfo['index'] = gpuIndex
gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"] gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment