#!/usr/bin/env bash
set -e 
BASE=/etc/ai-deployer
MINIO_URL=$(grep -E 'MINIO_URL' $BASE/ai.config|cut -d'=' -f2)
#目前支持RTX系列 30系 20系 10系 官方下载地址 https://us.download.nvidia.com/XFree86/Linux-x86_64/550.90.07/NVIDIA-Linux-x86_64-550.90.07.run
#Tesla系列 A100 V100 T4 P100 P40 官方下载地址 https://us.download.nvidia.com/tesla/550.90.07/NVIDIA-Linux-x86_64-550.90.07.run
#定义仓库下载地址
GENFORCE_URL=$MINIO_URL/ai-deployer/rtxseries/NVIDIA-Linux-x86_64-550.90.07.run
TESLA_URL=$MINIO_URL/ai-deployer/tesla/NVIDIA-Linux-x86_64-550.90.run
nvidia_driver_install(){
 case $1 in
 RXT-4090|RTX-4080|RTX-3090|RTX-3080|RTX-3070|RTX-3060|RTX-2080TI|RTX-2080|RTX-2070|RTX-2060)
    sudo service lightdm stop >/dev/null  2>&1
    cd /mnt
    rm -rf NVIDIA-Linux-x86_64-550.90.07.run
    sudo wget $GENFORCE_URL -q
    sudo chmod +x NVIDIA-Linux-x86_64-550.90.07.run
    sudo ./NVIDIA-Linux-x86_64-550.90.07.run --accept-license --silent --no-nouveau-check --disable-nouveau --no-opengl-files
    sudo service lightdm start >/dev/null  2>&1
    sudo nvidia-smi >/dev/null
   ;;
   A6000|A100|T4|V100|P40|P100)
    sudo service lightdm stop >/dev/null  2>&1
    cd /mnt
    rm -rf NVIDIA-Linux-x86_64-550.90.run
    sudo wget $TESLA_URL -q
    sudo chmod +x NVIDIA-Linux-x86_64-550.90.run
    sudo ./NVIDIA-Linux-x86_64-550.90.run --accept-license --silent --no-nouveau-check --disable-nouveau --no-opengl-files
    sudo service lightdm start >/dev/null  2>&1
    sudo nvidia-smi >/dev/null  2>&1
   ;;
 esac
}
  nvidia_driver_install $1