blob: 7bd1c540d44f9942ab746a77b500136f12bf1aa3 (
plain)
1
2
3
4
5
6
7
8
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
|
