ai-chatbot/check_status.exp
2026-05-23 11:09:03 +08:00

39 lines
880 B
Plaintext
Executable File

#!/usr/bin/expect
set timeout -1
set host "156.225.27.52"
set port "5204"
set user "root"
set password $env(DEPLOY_SSH_PASSWORD)
spawn ssh -p $port $user@$host "
echo '--- GPU Status ---'
nvidia-smi --query-gpu=name,memory.total,memory.free --format=csv,noheader
echo '--- Conda Environment ---'
source ~/miniconda3/bin/activate
conda env list | grep dreamomni2
echo '--- Installed Packages (Top 5) ---'
conda activate dreamomni2
pip list | grep -E 'torch|diffusers|transformers|accelerate|modelscope'
echo '--- Model Files ---'
ls -lh ~/ai-projects/DreamOmni2/checkpoints/DreamOmni2 | head -n 5
echo '--- Code Files ---'
ls -F ~/ai-projects/DreamOmni2
"
expect {
"password:" {
send "$password\r"
exp_continue
}
"yes/no" {
send "yes\r"
exp_continue
}
eof
}