tensorflow-setup.sh

# switch to python3
which python
sudo rm /usr/bin/python
sudo ln -s /usr/bin/python{3,}
python --version

# install python-dev
sudo apt install -y python3-dev

# install pip
wget https://bootstrap.pypa.io/get-pip.py
python get-pip.py --user
rm get-pip.py

# install python packages: matplotlib, scikit-learn, ipython
pip install --user matplotlib scikit-learn ipython

# install nvidia driver
sudo apt install -y nvidia-384

# download cuda
cd
mkdir cuda
cd cuda/
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
mv cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64{-deb,.deb}
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb
mv cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64{-deb,.deb}
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/2/cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64-deb
mv cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64{-deb,.deb}

# install cuda
sudo apt install -y ./cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt update
sudo apt install -y ./cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64.deb
sudo apt install -y ./cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64.deb
sudo apt install -y cuda

# download cudnn (via sftp)
cd
mkdir cudnn
mv cudnn-9.0-linux-x64-v7.1.tgz cudnn/
cd cudnn/

# install cudnn
tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h

# setup environemnt
cd
cat >> .bashrc << EOF
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
EOF
source .bashrc

# install tensorflow
pip install --user tensorflow-gpu

# verify tensorflow
cat > test.py << EOF
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
EOF
python test.py