55 lines
1.9 KiB
Bash
55 lines
1.9 KiB
Bash
#!/bin/bash
|
|
|
|
if [ "$EUID" -ne 0 ]
|
|
then echo "Please run as root"
|
|
exit
|
|
fi
|
|
|
|
#----------- Notes ------------------
|
|
#check for NVIDIA device:
|
|
#apt-get -y install pciutils
|
|
#lspci | grep VGA
|
|
#check if driver works with device:
|
|
#nvidia-smi
|
|
#------------------------------------
|
|
|
|
|
|
#Install drivers
|
|
#based on https://docs.nvidia.com/cuda/cuda-installation-guide-linux/
|
|
#and https://www.tensorflow.org/install/gpu
|
|
#Versions required: https://www.tensorflow.org/install/source#gpu
|
|
|
|
apt-get update
|
|
apt-get install linux-headers-$(uname -r)
|
|
|
|
#Set CUDA network repository to have priority
|
|
#wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
|
|
#mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
|
|
|
|
#Install the CUDA repository public GPG key
|
|
distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
|
|
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/7fa2af80.pub
|
|
add-apt-repository contrib
|
|
|
|
#Setup the CUDA network repository
|
|
add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/ /"
|
|
|
|
#Installs all CUDA Toolkit and Driver packages. Handles upgrading to the next version of the cuda package when it's released.
|
|
#Use the --no-install-recommends option for a lean driver install without any dependencies on X packages
|
|
apt-get update
|
|
apt-get install --no-install-recommends cuda-11-2
|
|
#apt-get install --no-install-recommendslibcudnn8=8.1.0.44-1+cuda11.2 libcudnn8-dev=8.1.0.44-1+cuda11.2
|
|
|
|
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
|
|
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8-dev_8.1.1.33-1+cuda11.2_amd64.deb
|
|
apt install ./libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
|
|
apt install ./libcudnn8-dev_8.1.1.33-1+cuda11.2_amd64.deb
|
|
|
|
|
|
|
|
#reboot system
|
|
#systemctl reboot
|
|
echo "Please reboot system"
|
|
|
|
|