#!/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