是否有兼容 Cuda 9.0 和 cudnn 7.1 的 tensorflow 版本
Posted
技术标签:
【中文标题】是否有兼容 Cuda 9.0 和 cudnn 7.1 的 tensorflow 版本【英文标题】:Is there a tensorflow version that is compatible with Cuda 9.0 and cudnn 7.1 【发布时间】:2018-12-31 20:14:21 【问题描述】:我有一台带有cuda 9.0
和cudnn 7.1
的机器。
我尝试在这台机器上使用tensorflow 1.7.0
,但它不起作用,因为这个版本的tensorflow
是为cudnn 7.0
创建的
在我的 gpu 上启动训练时出现此错误:
Loaded runtime CuDNN library: 7102 (compatibility version 7100) but source was compiled with 7005 (compatibility version 7000).
是否有与我的cuda
和cudnn
版本兼容的tensorflow
版本?我还需要这个工作 tensorflow 版本是>=1.7.0
。
我用谷歌搜索过这个问题,搜索了每个问题,但我从未得到cuda
和cudnn
的这些特定版本的答案。
【问题讨论】:
【参考方案1】:tensorflow_gpu-1.9.0
应该可以做到这一点。下面链接的表格显示了 CUDA 和 cuDNN 与不同版本的 tensorflow 的兼容性。
https://www.tensorflow.org/install/install_sources#tested_source_configurations
【讨论】:
此链接现已失效。在***.com/a/50622526/1825792 和tensorflow.org/install/source#tested_build_configurations 上有一个可用列表【参考方案2】:好的,我好像错过了一些安装步骤。
通过安装tensorflow
的最新版本,在撰写本文时它是1.9.0
,它确实可以在我的机器上运行。
【讨论】:
以上是关于是否有兼容 Cuda 9.0 和 cudnn 7.1 的 tensorflow 版本的主要内容,如果未能解决你的问题,请参考以下文章
深度学习服务器环境配置: Ubuntu17.04+Nvidia GTX 1080+CUDA 9.0+cuDNN 7.0+TensorFlow 1.3
深度学习 GPU环境 Ubuntu 16.04 + Nvidia GTX 1080 + Python 3.6 + CUDA 9.0 + cuDNN 7.1 + TensorFlow 1.6 环境配置(
Ubuntu16.04 安装配置 Caffe 过程 (GPU版+CUDA 9.0+cuDNN 9.0+OpenCV 3.4.1)