Merge pull request #58 from jaywonchung/better-gpu-usage

Support multi-gpu environment

scripts/gpu_usage.sh

@@ -16,7 +16,7 @@ get_platform()
 {
 	case $(uname -s) in
 		Linux)
-			gpu=$(lspci -v | grep -i gpu | grep driver | awk '{print $5}')
+			gpu=$(lspci -v | grep VGA | head -n 1 | awk '{print $5}')
 			echo $gpu
 		;;
 
@@ -32,10 +32,12 @@ get_platform()
 get_gpu()
 {
 	gpu=$(get_platform)
-	if [ $gpu == nvidia-gpu ]; then
+	if [[ "$gpu" == NVIDIA ]]; then
-		usage=$(nvidia-smi | grep "%" | awk '{print $13}')
+    usage=$(nvidia-smi | grep '%' | awk '{ sum += $13 } END { printf("%d%%\n", sum / NR) }')
-		echo $usage
+  else
+    usage='unknown'
 	fi
+  echo $usage
 }
 main()
 {
@@ -47,5 +49,3 @@ main()
 }
 # run the main driver
 main
-
-