Merge pull request #217 from microsoft/master

merge master

Merge pull request #217 from microsoft/master
merge master
a0846f2a · SparkSnail · GitHub · 9ce751de · 5845ca04 · a0846f2a
Unverified Commit a0846f2a authored Nov 22, 2019 by SparkSnail Committed by GitHub Nov 22, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/nni_gpu_tool/gpu_metrics_collector.py tools/nni_gpu_tool/gpu_metrics_collector.py +1 -1

No files found.
--- a/tools/nni_gpu_tool/gpu_metrics_collector.py
+++ b/tools/nni_gpu_tool/gpu_metrics_collector.py
@@ -35,7 +35,7 @@ def check_ready_to_run():
        pidList.remove(os.getpid())
        return not pidList
    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 = []
        for pid in pgrep_output.splitlines():
            pidList.append(int(pid))