Ubuntu 16.04 上未正确安装 tensorflow-gpu 和 cuDNN

tensorflow-gpu and cuDNN not properly installed on Ubuntu 16.04

我在 Ubuntu 16.04 (x86_64) 上安装了 CUDA 8.0,并在华硕 K501UX 笔记本电脑上安装了 NVidia Geforce GTX 950M(驱动程序 381)。 CUDA 安装后,根据 this doc,我可以 运行 deviceQuerybandwidthTest

下载的 cuDNN 5.1 是一个 .deb 包并自动安装。 但是/usr/local/cuda-8.0/include/里没有cudnn.h/usr/lib/x86_64-linux-gnu/目录里也没有libcudnn 表示 cuDNN 可能未正确安装。

Tensorflow-gpu 没有 运行,所以 tensorflowcuDNN 都有问题。任何想法表示赞赏。

已解决。关键是要保持 cuDNN 版本 与设备 兼容。我应该下载 cuDNN v5.1 library for Linux for .tar.gz format, and copy files to the corresponding directorys.