Skip to content

Instantly share code, notes, and snippets.

@hollisn

hollisn/Tensorflow-GPU Ubuntu 16.04, Cuda 8.0, CUDNN 5.1

Created January 18, 2017 02:01
Show Gist options
  • Save hollisn/ec5fd6e5205996025bee6ad49db97da3 to your computer and use it in GitHub Desktop.
Save hollisn/ec5fd6e5205996025bee6ad49db97da3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Tensorflow-GPU Ubuntu 16.04, Cuda 8.0, CUDNN 5.1
### Tensorflow
# cuda
# download cuda 8.0 and cd to the dir
sudo dpkg -i cuda-repo-ubuntu1604_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
# cuda toolkit (optional?)
sudo apt install nvidia-cuda-toolkit
nvcc --version
# cudnn
# download cudnn 5.1 and cd to the dir (need to register)
sudo tar -xvf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
# pip
sudo apt-get install python-pip python-dev
pip install --upgrade pip
pip install tensorflow-gpu
# add paths
vim .bash_profile
#export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
#export CUDA_HOME=/usr/local/cuda
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment