Initial commit

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