install_gpu.sh

#!/bin/bash

# Viet Tran, Miroslav Dobrucky IISAS,  2017
# Scripting for installing all necessary packages needed for GPU apps on vanilla Ubuntu image
# Make sure to choose cloud image with UEFI support 
# https://cloud-images.ubuntu.com/trusty/current/trusty-server-cloudimg-amd64-uefi1.img

# Installing gcc, make and kernel-extra 
sudo apt-get update
sudo apt-get -y install gcc make linux-image-extra-virtual

# Instaling NVIDA driver
wget http://us.download.nvidia.com/tesla/375.51/nvidia-driver-local-repo-ubuntu1404_375.51-1_amd64.deb
sudo dpkg -i nvidia-driver-local-repo-ubuntu1404_375.51-1_amd64.deb

# Installing CUDA. You can choose only runtime lib (cuda-runtime-version)
# or all packages  "sudo apt-get install cuda"

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda-runtime-8-0
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}