We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 85bd290 commit e02f038Copy full SHA for e02f038
userbenchmark/release-test/monitor_proc.sh
@@ -17,11 +17,11 @@ get_gpu_max_memory_usage_cuda() {
17
local max=$2
18
local curr
19
# Some processes might not use the GPU
20
- if ! nvidia-smi pmon -s m -c 1 -o T | grep "${my_pid}" >/dev/null 2>/dev/null; then
+ if ! nvidia-smi dmon -s m -c 1 -o T -i 0 | grep "${my_pid}" >/dev/null 2>/dev/null; then
21
echo "${max}"
22
return
23
fi
24
- curr=$(nvidia-smi pmon -s m -c 1 -o T | grep "${my_pid}" | awk '{print $5}' | sort | tail -1 | grep -o "[0-9.]*")
+ curr=$(nvidia-smi dmon -s m -c 1 -o T -i 0 | grep "${my_pid}" | awk '{print $4}' | sort | tail -1 | grep -o "[0-9.]*")
25
max "${curr}" "${max}"
26
}
27
0 commit comments