cudnn-install-tensorflow

REAME.md

cudnn install ubuntu + tensorflow

1. Install cuda
2. Export variables in ~/.bashrc or ~/.zshrc.

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

2. Download cudnn. Note: check TensorFlow install guide for the correct version.
3. Extract cudnn, this extracts a folder named cuda, cd into it.
4. Execute

sudo cp -P include/cudnn.h /usr/local/cuda/include
sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64

Uninstall

sudo apt-get remove cuda
suda apt-get purge cuda
sudo apt autoremove