Just 下载 cuDNN 5.1 并按照步骤操作(在 Ubuntu 16.04、CUDA 工具包 8.0 上测试)

$ tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz$ sudo cp -P cuda/include/cudnn.h/usr/local/cuda/include$ sudo cp -P cuda/lib64/libcudnn*/usr/local/cuda/lib64$ sudo chmod a+r/usr/local/cuda/include/cudnn.h/usr/local/cuda/lib64/libcudnn*

现在设置路径变量

$ vim ~/.bashrc导出 LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"导出 CUDA_HOME=/usr/local/cuda

完成

有关更多详细信息，您可以查看此网站

I encountered the following error when trying to run a TensorFlow program:

ImportError: libcudnn.Version: cannot open shared object file: No such file or director

Just download cuDNN 5.1 and follow the steps (Tested on Ubuntu 16.04, CUDA toolkit 8.0 )

$ tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
$ sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

Now set Path variables

$ vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

and done

For more details, you can check this site