tmux-kanagawa/scripts/gpu_usage.sh

49 lines
1.1 KiB
Bash
Raw Normal View History

2020-06-30 04:44:51 +02:00
#!/usr/bin/env bash
# setting the locale, some users have issues with different locales, this forces the correct one
export LC_ALL=en_US.UTF-8
2020-06-30 04:44:51 +02:00
current_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
source $current_dir/utils.sh
2020-06-30 04:44:51 +02:00
get_platform()
{
case $(uname -s) in
Linux)
gpu=$(lspci -v | grep VGA | head -n 1 | awk '{print $5}')
echo $gpu
;;
2020-06-30 04:44:51 +02:00
Darwin)
# TODO - Darwin/Mac compatability
;;
2020-06-30 04:44:51 +02:00
CYGWIN*|MINGW32*|MSYS*|MINGW*)
# TODO - windows compatability
;;
esac
2020-06-30 04:44:51 +02:00
}
2020-06-30 04:44:51 +02:00
get_gpu()
{
gpu=$(get_platform)
if [[ "$gpu" == NVIDIA ]]; then
2023-03-22 13:23:19 +01:00
usage=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | awk '{ sum += $0 } END { printf("%d%%\n", sum / NR) }')
2020-08-06 17:56:00 +02:00
else
usage='unknown'
fi
normalize_percent_len $usage
2020-06-30 04:44:51 +02:00
}
2020-06-30 04:44:51 +02:00
main()
{
# storing the refresh rate in the variable RATE, default is 5
RATE=$(get_tmux_option "@kanagawa-refresh-rate" 5)
gpu_label=$(get_tmux_option "@kanagawa-gpu-usage-label" "GPU")
gpu_usage=$(get_gpu)
echo "$gpu_label $gpu_usage"
sleep $RATE
2020-06-30 04:44:51 +02:00
}
2020-06-30 04:44:51 +02:00
# run the main driver
main