Try another field

userbenchmark/release-test/monitor_proc.sh

@@ -17,7 +17,7 @@ get_gpu_max_memory_usage_cuda() {
     local max=$2
     local curr
     # pick the process that uses the most GPU memory
-    curr=$(nvidia-smi dmon -s m -c 1 -o T -i 0 | tail -n +3 | awk '{print $3}' | sort -n | tail -1 | grep -o "[0-9.]*")
+    curr=$(nvidia-smi dmon -s m -c 1 -o T -i 0 | tail -n +3 | awk '{print $2}' | sort -n | tail -1 | grep -o "[0-9.]*")
     # Some processes might not use the GPU, then memory usage should be 0
     if [ "${curr}" -eq 0 ] ; then
         echo "${max}"