tmux-kanagawa/scripts/gpu_usage.sh

38 lines
580 B
Bash
Raw Normal View History

2020-06-30 04:44:51 +02:00
#!/usr/bin/env bash
get_platform()
{
case $(uname -s) in
Linux)
2020-08-06 17:56:00 +02:00
gpu=$(lspci -v | grep VGA | head -n 1 | awk '{print $5}')
2020-06-30 04:44:51 +02:00
echo $gpu
;;
Darwin)
# TODO - Darwin/Mac compatability
;;
CYGWIN*|MINGW32*|MSYS*|MINGW*)
# TODO - windows compatability
;;
esac
}
get_gpu()
{
gpu=$(get_platform)
2020-08-06 17:56:00 +02:00
if [[ "$gpu" == NVIDIA ]]; then
usage=$(nvidia-smi | grep '%' | awk '{ sum += $13 } END { printf("%d%%\n", sum / NR) }')
else
usage='unknown'
2020-06-30 04:44:51 +02:00
fi
2020-08-06 17:56:00 +02:00
echo $usage
2020-06-30 04:44:51 +02:00
}
main()
{
gpu_usage=$(get_gpu)
echo "GPU $gpu_usage"
sleep 10
}
# run the main driver
main