51 lines
1.7 KiB
Bash
51 lines
1.7 KiB
Bash
|
#!/bin/bash
|
|||
|
|
set -eo pipefail
|
|||
|
|
LOG_DIR="{{ log_base_dir }}/{{ driver.name }}" # 从角色变量注入路径
|
|||
|
|
LOG_FILE="${LOG_DIR}/install-$(date +%Y%m%d).log"
|
|||
|
|
mkdir -p "$LOG_DIR"
|
|||
|
|
exec > >(tee -a "$LOG_FILE") 2>&1
|
|||
|
|
|
|||
|
|
# 参数解析（严格匹配角色定义的操作）
|
|||
|
|
OPERATION=""
|
|||
|
|
VERSION=""
|
|||
|
|
while [[ $# -gt 0 ]]; do
|
|||
|
|
case "$1" in
|
|||
|
|
--install) OPERATION="install" ;;
|
|||
|
|
--uninstall) OPERATION="uninstall" ;;
|
|||
|
|
--version) VERSION="$2"; shift ;;
|
|||
|
|
*) echo "错误：未知参数 $1" >&2; exit 1 ;;
|
|||
|
|
esac
|
|||
|
|
shift
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
# 安装逻辑（使用角色专属变量）
|
|||
|
|
install() {
|
|||
|
|
local DRIVER_VERSION="${VERSION:-$DEFAULT_VERSION}"
|
|||
|
|
echo "[$(date)] 开始安装NVIDIA驱动（版本：$DRIVER_VERSION，型号：$GPU_MODEL）..."
|
|||
|
|
|
|||
|
|
# 企业内部镜像下载（安全加速）
|
|||
|
|
wget -q "${DOWNLOAD_URL}/${DRIVER_VERSION}/NVIDIA-Linux-x86_64-${DRIVER_VERSION}.run" \
|
|||
|
|
-O "/tmp/nvidia-driver.run"
|
|||
|
|
|
|||
|
|
# 静默安装（企业级无交互模式）
|
|||
|
|
sh "/tmp/nvidia-driver.run" --silent --no-x-check --no-nouveau-check
|
|||
|
|
|
|||
|
|
# 严格功能验证（硬件型号匹配）
|
|||
|
|
if ! nvidia-smi --query-gpu=name --format=csv,noheader | grep -q "$GPU_MODEL"; then
|
|||
|
|
echo "错误：驱动安装后未识别到目标GPU型号" >&2; exit 1
|
|||
|
|
fi
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 卸载逻辑（幂等性设计）
|
|||
|
|
uninstall() {
|
|||
|
|
echo "[$(date)] 开始卸载NVIDIA驱动..."
|
|||
|
|
/usr/bin/nvidia-uninstall --silent # 官方静默卸载工具
|
|||
|
|
rm -f "/tmp/nvidia-driver.run" # 清理残留文件
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 主流程（依赖角色变量注入的默认值）
|
|||
|
|
DEFAULT_VERSION="{{ driver.default_version }}"
|
|||
|
|
DOWNLOAD_URL="{{ driver.download_url }}"
|
|||
|
|
install || uninstall # 根据OPERATION执行对应函数（由剧本参数控制）
|
|||
|
|
exit 0
|