Scribd
Upload
104 views

and Install Ubuntu 14.04 or 16.04 Install CUDA 8: Instantly Share Code, Notes, and Snippets

This document provides instructions for installing CUDA and Tensorflow on Ubuntu 14.04 or 16.04. It describes downloading the appropriate versions of CUDA and CUDNN and configuring environment variables to enable GPU support in Tensorflow. The instructions guide the user through installing CUDA and necessary drivers, then CUDNN library files before installing Tensorflow from pip using a GPU-enabled package URL.

Uploaded by

Soma Sundar
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
104 views

and Install Ubuntu 14.04 or 16.04 Install CUDA 8: Instantly Share Code, Notes, and Snippets

This document provides instructions for installing CUDA and Tensorflow on Ubuntu 14.04 or 16.04. It describes downloading the appropriate versions of CUDA and CUDNN and configuring environment variables to enable GPU support in Tensorflow. The instructions guide the user through installing CUDA and necessary drivers, then CUDNN library files before installing Tensorflow from pip using a GPU-enabled package URL.

Uploaded by

Soma Sundar
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 2

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

You might also like