Shell:批量获取机器配置信息的脚本

批量获取机器配置信息的脚本

#!/usr/bin/env bash

#######################################
HOSTS=(

)
passwd=
#######################################

function check_sshpass_install(){
dpkg -s sshpass | grep -w "installed" &> /dev/null
result_sshpass=$(echo $?)
[[ ${result_sshpass} != "0" ]] && apt-get -y install sshpass &> /dev/null
}

check_sshpass_install

echo "本次操作机器共${#HOSTS[@]}台"
for host in ${HOSTS[@]};do
echo "============${host}正在处理==========="
## 获取cpu型号
cpu_type=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "lscpu"|awk '/Model name/{print $3,$4,$5,$6}')

## 获取内核版本
kernel_version=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "uname -r")

## 获取系统版本
sys_version=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "uname -v"|awk -F"[~| ]" '{print $2}')

## 获取GPU版本
gpu_version=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "nvidia-smi"|awk '/Driver Version/{print $6}')
## 获取GPU类型
gpu_type=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "nvidia-smi -L"|awk '{print $4,$5,$6}'|head -1)
## 获取GPU数量
gpu_num=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "nvidia-smi -L"|awk '{print $4,$5,$6}'|wc -l)

## 获取磁盘信息
disk_info=$(sshpass -p ${passwd} ssh -o "StrictHostKeyChecking no" ${host} "lsblk")
echo "${host},${cpu_type},${kernel_version},${sys_version},${gpu_version},${gpu_type},${gpu_num}" >> server_info.csv
echo -e "${host}\n${disk_info}" >> disk_info.csv
done