Install Cuda on Ubuntu 18.04

install_cuda_repo.sh

#!/bin/bash

sudo apt purge nvidia*
sudo apt autoremove
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub # replace <version> with the correct one
sudo apt update
sudo apt install cuda
sudo apt install nvidia-cuda-toolkit