Toggle between using GPU for Deep Learning and Video Games.

gpu_toggle.sh

#!/bin/bash
flag=$1
if [[ $flag == 'DL' ]]; then
	# -------------
	# Deep Learning
	# -------------
	echo "Configuring GPU for Deep Learning..."
	sudo apt-get install -y libnvidia-common-440
	sudo apt-get install -y cuda
	sudo apt-get install -y cuda-drivers
	sudo apt-get install -y nvidia-driver-418
	sudo apt-get install -y cuda
	sudo apt-get install -y libcupti-dev
	python3 -m pip install --user --upgrade tensorflow-gpu
	python3 -m pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
	echo "Now we reboot..."
	reboot
elif [[ $flag == 'VG' ]]; then
	# -----------
	# Video games
	# -----------
	echo "Configuring GPU for Video Games..."
	sudo apt-get install -y libnvidia-compute-440
	sudo apt-get install -y libnvidia-gl-440
	sudo ubuntu-drivers autoinstall
	echo "Now we reboot..."
	reboot
else
	echo "Don't forget to supply either 'DL' or 'VG' as an additional argument."
fi