Fix #3: Correct GPU utilisation

This commit is contained in:
Miquel Massot 2023-07-04 17:16:17 +01:00 committed by GitHub
parent 139b9754db
commit f40f0205d5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -25,7 +25,7 @@ get_platform() {
get_gpu() { get_gpu() {
gpu=$(get_platform) gpu=$(get_platform)
if [[ "$gpu" == NVIDIA ]]; then if [[ "$gpu" == NVIDIA ]]; then
usage=$(nvidia-smi | grep '%' | awk '{ sum += $13 } END { printf("%d%%\n", sum / NR) }') usage=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | awk '{ sum += $0 } END { printf("%s%%\n", sum / NR) }')
else else
usage='unknown' usage='unknown'
fi fi