diff --git a/install-nvidia.sh b/install-nvidia.sh new file mode 100644 index 0000000..b764225 --- /dev/null +++ b/install-nvidia.sh @@ -0,0 +1,58 @@ +#!/bin/bash + +RED='\033[0;31m' +RESET='\033[0m' +KERNEL="$(uname -r)" + +function rootCheck(){ + if [ "$USER" != "root" ]; then + echo "$RED You need to be root to run this script... $RESET" + exit 1 + fi +} + +function xorgInstall(){ + pacman -S xorg xorg-server + + if [ "$?" != "0" ]; then + echo "$RED An error occurred while installing Xorg $RESET" + exit 1 + fi +} + +function nvidiaInstall(){ + pacman -S nvidia nvidia-utils nvidia-settings opencl-nvidia + + if [ "$?" != "0" ]; then + echo "$RED An error occurred while installing the NVIDIA drivers on kernel $KERNEL $RESET" + exit 1 + fi + + cat /usr/lib/modprobe.d/nvidia-lts.conf +} + +function nvidiaInstallLTS(){ + pacman -S nvidia-lts nvidia-utils nvidia-settings opencl-nvidia + + if [ "$?" != "0" ]; then + echo "$RED An error occurred while installing the NVIDIA LTS drivers on kernel $KERNEL LTS $RESET" + exit 1 + fi + + cat /usr/lib/modprobe.d/nvidia-lts.conf +} + +rootCheck +xorgInstall + +case $1 in + + --lts) + nvidiaInstallLTS + ;; + *) + nvidiaInstall + ;; +esac + +exit 0 \ No newline at end of file