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

Question

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

下载的 cuDNN 5.1 是 .deb打包并自动安装。 但是，没有cudnn.h在 /usr/local/cuda-8.0/include/ , 没有 libcudnn在 /usr/lib/x86_64-linux-gnu/目录。 this validation表明 cuDNN 可能未正确安装。

Tensorflow-gpu 没有运行，所以要么 tensorflow或 cuDNN是有问题的。任何想法表示赞赏。

Answer 1

已解决。关键是要保持 cuDNN 版本与设备兼容。我应该下载 .tar.gz 格式的适用于 Linux 的 cuDNN v5.1 库，并将文件复制到相应的目录中。