From:https://gist.github.com/ashokpant/5c4e9481615f54af4025ab2085f85869
Thanks.
#!/bin/bash
install CUDA Toolkit v9.0
instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb)
CUDA_REPO_PKG=“cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb”
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/${CUDA_REPO_PKG}
sudo dpkg -i ${CUDA_REPO_PKG}
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda-9-0
CUDA_PATCH1=“cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb”
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/${CUDA_PATCH1}
sudo dpkg -i ${CUDA_PATCH1}
sudo apt-get update
install cuDNN v7.0
CUDNN_PKG=“libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb”
wget https://github.com/ashokpant/cudnn_archive/raw/master/v7.0/${CUDNN_PKG}
sudo dpkg -i ${CUDNN_PKG}
sudo apt-get update
install NVIDIA CUDA Profile Tools Interface ( libcupti-dev v9.0)
sudo apt-get install cuda-command-line-tools-9-0
set environment variables
export PATH=
P
A
T
H
:
/
u
s
r
/
l
o
c
a
l
/
c
u
d
a
−
9.0
/
b
i
n
e
x
p
o
r
t
C
U
D
A
H
O
M
E
=
{PATH}:/usr/local/cuda-9.0/bin export CUDA_HOME=
PATH:/usr/local/cuda−9.0/binexportCUDAHOME={CUDA_HOME}:/usr/local/cuda:/usr/local/cuda-9.0
export LD_LIBRARY_PATH=
L
D
L
I
B
R
A
R
Y
P
A
T
H
:
/
u
s
r
/
l
o
c
a
l
/
c
u
d
a
−
9.0
/
l
i
b
64
e
x
p
o
r
t
L
D
L
I
B
R
A
R
Y
P
A
T
H
=
{LD_LIBRARY_PATH}:/usr/local/cuda-9.0/lib64 export LD_LIBRARY_PATH=
LDLIBRARYPATH:/usr/local/cuda−9.0/lib64exportLDLIBRARYPATH=LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64