Skip to content

GitLab

Unverified Commit a0846f2a authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Merge pull request #217 from microsoft/master 

merge master
parents 9ce751de 5845ca04
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/nni_gpu_tool/gpu_metrics_collector.py
View file @ a0846f2a
...@@ -35,7 +35,7 @@ def check_ready_to_run(): ...@@ -35,7 +35,7 @@ def check_ready_to_run():
pidList.remove(os.getpid()) pidList.remove(os.getpid())
return not pidList return not pidList
else: else:
pgrep_output = subprocess.check_output('pgrep -fx \'python3 -m nni_gpu_tool.gpu_metrics_collector\'', shell=True) pgrep_output = subprocess.check_output('pgrep -fxu "$(whoami)" \'python3 -m nni_gpu_tool.gpu_metrics_collector\'', shell=True)
pidList = [] pidList = []
for pid in pgrep_output.splitlines(): for pid in pgrep_output.splitlines():
pidList.append(int(pid)) pidList.append(int(pid))
......
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