diff options
Diffstat (limited to '.script/gpu_usage_nvidia.sh')
| -rwxr-xr-x | .script/gpu_usage_nvidia.sh | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/.script/gpu_usage_nvidia.sh b/.script/gpu_usage_nvidia.sh new file mode 100755 index 0000000..7bd1c54 --- /dev/null +++ b/.script/gpu_usage_nvidia.sh @@ -0,0 +1,9 @@ +#!/bin/sh + +out=$(nvidia-smi -i 0 --query-gpu=utilization.gpu,memory.used --format=csv,noheader,nounits 2>/dev/null) || exit 0 + +gpu=$(printf "%s" "$out" | awk -F',' '{gsub(/ /,"",$1); print $1}') +mem=$(printf "%s" "$out" | awk -F',' '{gsub(/ /,"",$2); print $2}') + +# mem is in MiB from nvidia-smi; display as M for brevity. +printf "VRAM %sM|GPU %s%%\n" "$mem" "$gpu" > /tmp/gpu_usage |