42 lines
1.1 KiB
Bash
42 lines
1.1 KiB
Bash
|
|
#!/bin/bash
|
||
|
|
|
||
|
|
IDLE_TIMEOUT_MINUTES=20
|
||
|
|
|
||
|
|
# Check 1: Any logged-in users?
|
||
|
|
if who | grep -qv "tty"; then
|
||
|
|
echo "Active user sessions found. Skipping suspend."
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check 2: CPU idle percentage
|
||
|
|
CPU_IDLE=$(mpstat 1 1 | tail -n 1 | awk '/:/ {print $12}' | sed 's/,/./g')
|
||
|
|
CPU_THRESHOLD=90
|
||
|
|
|
||
|
|
if (( $(echo "$CPU_IDLE < $CPU_THRESHOLD" | bc -l) )); then
|
||
|
|
echo "CPU too active. Skipping suspend."
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
if ! command -v nvidia-smi &> /dev/null; then
|
||
|
|
echo "nvidia-smi not found, skipping GPU check"
|
||
|
|
else
|
||
|
|
# Check 3: Is GPU idle?
|
||
|
|
GPU_UTIL=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | head -n 1)
|
||
|
|
GPU_THRESHOLD=10
|
||
|
|
|
||
|
|
if (( GPU_UTIL > GPU_THRESHOLD )); then
|
||
|
|
echo "GPU is busy (utilization=${GPU_UTIL}%). Skipping suspend."
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check 4: Is OLLAMA active?
|
||
|
|
OLLAMA_ACTIVE=$(nvidia-smi --query-compute-apps=name,used_memory --format=csv,noheader,nounits | grep ollama | wc -l)
|
||
|
|
if (( OLLAMA_ACTIVE > 0 )); then
|
||
|
|
echo "OLLMAMA is active. Skipping suspend."
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "System idle. Suspending..."
|
||
|
|
shutdown now
|