Install CUDA 10 on Ubuntu 18.04

install-cuda-10-bionic.sh

#!/bin/bash

# Purge existign CUDA first
sudo apt --purge remove "cublas*" "cuda*"
sudo apt --purge remove "nvidia*"

# Install CUDA Toolkit 10
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub && sudo apt update
sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

sudo apt update
sudo apt install -y cuda

# Install CuDNN 7 and NCCL 2
wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo dpkg -i nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb

sudo apt update
sudo apt install -y libcudnn7 libcudnn7-dev libnccl2 libc-ares-dev

sudo apt autoremove

# NOTE: if want to change cuda version, use aptitude to manual install cuda

# Link libraries to standard locations
# ll /usr/local/
# /usr/local/cuda -> /usr/local/cuda-10.1/
sudo mkdir -p /usr/local/cuda/nccl/lib
sudo ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/nccl/lib/
sudo ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7 /usr/local/cuda/lib64/

# ~/.bashrc add lines
export PATH=/usr/local/cuda/bin:$PATH
export CUDADIR=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

echo 'If everything worked fine, reboot now.'

# for cuda 10-2
# pushd /usr/local/cuda-10.2/lib64
# sudo ln -s libcudart.so.10.2.89 libcudart.so.10.0
# sudo ln -s libcufft.so.10.1.2.89 libcufft.so.10.0
# sudo ln -s libcurand.so.10.1.2.89 libcurand.so.10.0
# sudo ln -s libcusolver.so.10.3.0.89 libcusolver.so.10.0
# sudo ln -s libcusparse.so.10.3.1.89 libcusparse.so.10.0
# sudo ln -s /usr/lib/x86_64-linux-gnu/libcublas.so.10 /usr/local/cuda-10.2/lib64/libcublas.so.10.0
# popd