Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
c71dadba
"fs/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "5e20b170a783c8a476991ca4a7ee32029761fe66"
Unverified
Commit
c71dadba
authored
Apr 06, 2021
by
SparkSnail
Committed by
GitHub
Apr 06, 2021
Browse files
Add log information for gpu collector (#3506)
parent
d6b5c6ca
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
ts/nni_manager/training_service/common/gpuData.ts
ts/nni_manager/training_service/common/gpuData.ts
+2
-1
ts/nni_manager/training_service/common/util.ts
ts/nni_manager/training_service/common/util.ts
+2
-1
No files found.
ts/nni_manager/training_service/common/gpuData.ts
View file @
c71dadba
...
@@ -79,6 +79,7 @@ export function parseGpuIndices(gpuIndices?: string): Set<number> | undefined {
...
@@ -79,6 +79,7 @@ export function parseGpuIndices(gpuIndices?: string): Set<number> | undefined {
export
const
GPU_INFO_COLLECTOR_FORMAT_WINDOWS
:
string
=
export
const
GPU_INFO_COLLECTOR_FORMAT_WINDOWS
:
string
=
`
`
$env:METRIC_OUTPUT_DIR="{0}"
$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
Write $app.ID | Out-File {1} -NoNewline -encoding utf8
`
;
`
;
ts/nni_manager/training_service/common/util.ts
View file @
c71dadba
...
@@ -236,7 +236,8 @@ export function getScriptName(fileNamePrefix: string): string {
...
@@ -236,7 +236,8 @@ export function getScriptName(fileNamePrefix: string): string {
}
}
export
function
getGpuMetricsCollectorBashScriptContent
(
scriptFolder
:
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
{
export
function
runGpuMetricsCollector
(
scriptFolder
:
string
):
void
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment