Install NVIDIA drivers and CUDA for Ubuntu 18.04 LTS

easy_install_cuda_ubuntu-1804.sh

CUDA_REPO_PKG=cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64

wget -O /tmp/${CUDA_REPO_PKG} ${CUDA_URL}/${CUDA_REPO_PKG} 

sudo dpkg -i /tmp/${CUDA_REPO_PKG}

sudo apt-key adv --fetch-keys ${CUDA_URL}/7fa2af80.pub 

rm -f /tmp/${CUDA_REPO_PKG}

sudo apt-get update

sudo apt-get install cuda-drivers

install_cuda_ubuntu-1804.sh

# https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
# https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/


# ========================== FROM FILES ==========================

# DOwnload NVIDIA drivers (installing from run file to avoid openGL libraries: https://www.geforce.com/drivers)
wget http://us.download.nvidia.com/XFree86/Linux-x86_64/440.82/NVIDIA-Linux-x86_64-440.82.run
sudo chmod +x NVIDIA-Linux-x86_64-440.82.run

# Download CUDA (find your version at: https://developer.nvidia.com/cuda-downloads)
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo chmod +x cuda_10.2.89_440.33.01_linux.run

# Uninstall old stuff
sudo apt-get --purge remove nvidia-*
sudo nvidia-uninstall

# Reboot and CTRL+ALT+F1 and log in
sudo service lightdm stop

# install the driver
sudo ./NVIDIA-Linux-x86_64-440.82.run --no-opengl-files
# install CUDA
sudo ./cuda_10.2.89_440.33.01_linux.run --no-opengl-libs

# Verify installation
nvidia-smi
cat /proc/driver/nvidia/version


# ========================== FROM PACKAGE REPOSITORIES ==========================
# Add NVIDIA package repositories
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo apt-get update
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update

# Install NVIDIA driver
sudo apt-get install --no-install-recommends nvidia-driver-430
# Reboot. Check that GPUs are visible using the command: nvidia-smi

# Install development and runtime libraries (~4GB)
sudo apt-get install --no-install-recommends \
    cuda-10-1 \
    libcudnn7=7.6.4.38-1+cuda10.1  \
    libcudnn7-dev=7.6.4.38-1+cuda10.1


# Install TensorRT. Requires that libcudnn7 installed as per above.
sudo apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 \
    libnvinfer-dev=6.0.1-1+cuda10.1 \
    libnvinfer-plugin6=6.0.1-1+cuda10.1