python - Tensorflow 0.7.1 与 Cuda 工具包 7.5 和 cuDNN 7.0

Question

我最近尝试将我的 Tensorflow 安装从 0.6 升级到 0.7.1(Ubuntu 15.10、Python 2.7)，因为它被描述为与更多最新的 Cuda 库兼容。一切都运行良好，包括来自 Tensorflow 入门页面的简单测试。但是我无法使用 cuDNN。使用 cuDNN 运行程序时，我首先收到警告

“无法加载 cuDNN DSO”

后来程序崩溃了

I tensorflow/core/common_runtime/gpu/gpu_device.cc:717] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX 980, pci bus id: 0000:01:00.0)
I tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:73] Allocating 3.30GiB bytes.
I tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:83] GPU 0 memory begins at 0x704a80000 extends to 0x7d80c8000
F tensorflow/stream_executor/cuda/cuda_dnn.cc:204] could not find cudnnCreate in cudnn DSO; dlerror: /usr/local/lib/python2.7/dist-packages/tensorflow/python/_pywrap_tensorflow.so: undefined symbol: cudnnCreate

我为 Cuda 安装下载的文件是

• cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb和
• cudnn-7.0-linux-x64-v4.0-prod.tgz

我按照 Tensorflow getting started page 上的说明进行操作除了使用 cuDNN 7.0 而不是 6.5。 $LD_LIBRARY_PATH 是“/usr/local/cuda/lib64”

我不知道为什么找不到 cudnnCreate。有没有人已经成功安装了这个配置并可以给我建议？

Answer 1

当我忘记设置 LD_LIBRARY_PATH 和 CUDA_HOME 环境变量时，我得到了同样的错误:

导出 LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"导出 CUDA_HOME=/usr/local/cuda