diff --git a/ts/nni_manager/training_service/common/gpuData.ts b/ts/nni_manager/training_service/common/gpuData.ts index 8767baf6e42..c20a6926867 100644 --- a/ts/nni_manager/training_service/common/gpuData.ts +++ b/ts/nni_manager/training_service/common/gpuData.ts @@ -79,6 +79,7 @@ export function parseGpuIndices(gpuIndices?: string): Set | undefined { export const GPU_INFO_COLLECTOR_FORMAT_WINDOWS: string = ` $env:METRIC_OUTPUT_DIR="{0}" -$app = Start-Process "python" -ArgumentList "-m nni.tools.gpu_tool.gpu_metrics_collector" -passthru -NoNewWindow +$app = Start-Process "python" -ArgumentList "-m nni.tools.gpu_tool.gpu_metrics_collector" -passthru -NoNewWindow \ +-redirectStandardOutput {0}\\stdout -redirectStandardError {0}\\stderr Write $app.ID | Out-File {1} -NoNewline -encoding utf8 `; diff --git a/ts/nni_manager/training_service/common/util.ts b/ts/nni_manager/training_service/common/util.ts index 791d7dcebbf..c88c9cd957b 100644 --- a/ts/nni_manager/training_service/common/util.ts +++ b/ts/nni_manager/training_service/common/util.ts @@ -236,7 +236,8 @@ export function getScriptName(fileNamePrefix: string): string { } export function getGpuMetricsCollectorBashScriptContent(scriptFolder: string): string { - return `echo $$ > ${scriptFolder}/pid ; METRIC_OUTPUT_DIR=${scriptFolder} python3 -m nni.tools.gpu_tool.gpu_metrics_collector`; + return `echo $$ > ${scriptFolder}/pid ; METRIC_OUTPUT_DIR=${scriptFolder} python3 -m nni.tools.gpu_tool.gpu_metrics_collector \ +1>${scriptFolder}/stdout 2>${scriptFolder}/stderr`; } export function runGpuMetricsCollector(scriptFolder: string): void {