Skip to content

Commit

Permalink
hotfix. Healtcheck pass if GPU usage stricly 0
Browse files Browse the repository at this point in the history
  • Loading branch information
damienlaine committed Nov 20, 2024
1 parent d3bc76d commit e61c4c0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion healthcheck.sh
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ else
fi

# Check if GPU is in use
has_gpu=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | grep -v '^0$')
has_gpu=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | grep -v '^0$' | wc -l)
if [ "$has_gpu" -gt 0 ]; then
echo "HealthCheck PASS: GPU is being utilized, marking service as healthy."
exit 0
Expand Down

0 comments on commit e61c4c0

Please sign in to comment.