tmux-kanagawa/scripts/gpu_usage.sh
Ethan Edwards 7f8bb35543
Merge pull request #58 from jaywonchung/better-gpu-usage
Support multi-gpu environment
2020-09-03 15:11:04 -04:00

51 lines
950 B
Bash
Executable file

#!/usr/bin/env bash
# function for getting the refresh rate
get_tmux_option() {
local option=$1
local default_value=$2
local option_value=$(tmux show-option -gqv "$option")
if [ -z $option_value ]; then
echo $default_value
else
echo $option_value
fi
}
get_platform()
{
case $(uname -s) in
Linux)
gpu=$(lspci -v | grep VGA | head -n 1 | awk '{print $5}')
echo $gpu
;;
Darwin)
# TODO - Darwin/Mac compatability
;;
CYGWIN*|MINGW32*|MSYS*|MINGW*)
# TODO - windows compatability
;;
esac
}
get_gpu()
{
gpu=$(get_platform)
if [[ "$gpu" == NVIDIA ]]; then
usage=$(nvidia-smi | grep '%' | awk '{ sum += $13 } END { printf("%d%%\n", sum / NR) }')
else
usage='unknown'
fi
echo $usage
}
main()
{
# storing the refresh rate in the variable RATE, default is 5
RATE=$(get_tmux_option "@dracula-refresh-rate" 5)
gpu_usage=$(get_gpu)
echo "GPU $gpu_usage"
sleep $RATE
}
# run the main driver
main