小编Pet*_*r W的帖子

使用Cuda Toolkit 7.5和cuDNN 7.0的Tensorflow 0.7.1

我最近尝试将我的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
Run Code Online (Sandbox Code Playgroud)

我为Cuda安装下载的文件是

  • cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb和
  • cudnn-7.0的Linux-x64的V4.0-prod.tgz

我按照Tensorflow入门页面上的说明进行操作,但使用cuDNN 7.0而不是6.5.$ LD_LIBRARY_PATH是"/ usr/local/cuda/lib64"

我不知道为什么找不到cudnnCreate.是否有人成功安装了此配置并可以给我建议?

python linux tensorflow

7
推荐指数
2
解决办法
5713
查看次数

标签 统计

linux ×1

python ×1

tensorflow ×1