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