使用Cuda Toolkit 7.5和cuDNN 7.0的Tensorflow 0.7.1 [英] Tensorflow 0.7.1 with Cuda Toolkit 7.5 and cuDNN 7.0

问题描述

我最近尝试将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]创建TensorFlow设备(/gpu:0)->(设备:0，名称:GeForce GTX 980，PCI总线ID:0000:01:00.0)我tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:73]分配了3.30GiB字节.我tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:83] GPU 0内存从0x704a80000开始扩展到0x7d80c8000F tensorflow/stream_executor/cuda/cuda_dnn.cc:204]在cudnn DSO中找不到cudnnCreate；dlerror:/usr/local/lib/python2.7/dist-packages/tensorflow/python/_pywrap_tensorflow.so:未定义符号:cudnnCreate 

我为Cuda安装下载的文件是

• cuda-repo-ubuntu1504-7-5--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.是否有人成功安装了此配置并可以给我建议?

解决方案

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

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

I recently tried to upgrade my Tensorflow installation from 0.6 to 0.7.1 (Ubuntu 15.10, Python 2.7) because it is described to be compatible with more up-to-date Cuda libraries. Everything works well including the simple test from the Tensorflow getting started page. However I'm not able to use cuDNN. When running a program using cuDNN, I first get a warning

"Unable to load cuDNN DSO"

and later the program crashes with

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

The files I downloaded for the Cuda Installation were

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

I followed the instructions on the Tensorflow getting started page with the exception of using cuDNN 7.0 instead of 6.5. $LD_LIBRARY_PATH is "/usr/local/cuda/lib64"

I have no clue why cudnnCreate is not found. Is there somebody who has successfully installed this configuration and can give me advice?

解决方案

I get the same error when I forgot to set the LD_LIBRARY_PATH and CUDA_HOME environment variables:

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

这篇关于使用Cuda Toolkit 7.5和cuDNN 7.0的Tensorflow 0.7.1的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！