#!/bin/bash
set -e

echo "安装 NVIDIA 驱动 {{ gpu.driver_version }}（离线方式）"

bash NVIDIA-Linux-x86_64-{{ gpu.driver_version }}.run --silent

echo "加载 nvidia 模块"
modprobe nvidia
modprobe nvidia_uvm

echo "NVIDIA 驱动安装完成"