11/30/2017 Installation of CUDA & Tensorflow in Ubuntu 14.04 or 16.

04 · GitHub

Instantly share code, notes, and snippets. Create a gist now

vbalnt / cuda-ubuntu.md
Last active 11 days ago

Installation of CUDA & Tensorflow in Ubuntu 14.04 or 16.04

cuda-ubuntu.md

This document describes how to install the combination of 14.04 + CUDA 7.5 + Tensorflow. This combination is the easiest to
install without anything like compilation from sources etc.

Download and install Ubuntu 14.04 or 16.04

http://releases.ubuntu.com/14.04/ http://releases.ubuntu.com/16.04/

Install CUDA 8
This way below installs CUDA and all the related things (e.g. drivers needed).

Go to https://developer.nvidia.com/cuda-downloads
Choose Linux -> x86_64 -> Ubuntu -> 14.04 or 16.04 -> deb (local)
Press download

After download is over, open a terminal and navigate to Downloads .

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb

sudo apt-get update
sudo apt-get install cuda

the dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb might differ (check the download name - or use tab for
autocomplete)

Restart the PC to activate CUDA + the new drivers.

Install CUDNN 5.1

CUDNN makes CNNs faster with some convolutional and other optimizations.

Navigate to https://developer.nvidia.com/cudnn and register for an account (it's free). After you make your account, login and
go to downloads.

Choose the following Download cuDNN v5.1 (August 10, 2016), for CUDA 8 cuDNN v5.1 Library for Linux

after it is downloaded navigate to Downloads and extract the tar file. You will get a folder called cuda . Open a terminal and
run the following to navigate to this folder and put the cudnn files to your system folders.

cd ~/Downloads/cuda
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/

https://gist.github.com/vbalnt/a0f789d788a99bfb62b61cb809246d64 1/2
11/30/2017 Installation of CUDA & Tensorflow in Ubuntu 14.04 or 16.04 · GitHub

Update your .bashrc

Open .bashrc from your home folder with an editor (the dot indicates that it is a hidden file). a way to do this for example is

gedit ~/.bashrc

Add the following lines to the end of the file.

# add cuda tools to command path

export PATH=/usr/local/cuda/bin:${PATH}
export MANPATH=/usr/local/cuda/man:${MANPATH}

# add cuda libraries to library path

if [[ "${LD_LIBRARY_PATH}" != "" ]]
then
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
else
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
fi

Install tensorflow
Open a terminal and write

export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc2-cp27-none-linux_x86_64.
sudo pip install --upgrade $TF_BINARY_URL

Enjoy your new gpu-enabled deep learning setup :)

https://gist.github.com/vbalnt/a0f789d788a99bfb62b61cb809246d64 2/2