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
|