无法使用GPU运行Tensorflow/Keras

Question

我尝试使用GPU运行Keras，但出现以下错误：

C:\Python36\lib\site-packages\skimage\transform_warps.py:84: UserWarning: 默认模式“constant”将在skimage 0.15中更改为“reflect”。 warn("The default mode, 'constant', will be changed to 'reflect' in " E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_dnn.cc:378]

载入的CuDNN运行库：7102（兼容版本7100），但源代码是用7003（兼容版本7000）编译的。如果使用二进制安装，请将CuDNN库升级到相匹配的版本。如果从源码构建，请确保在运行时加载的库与编译配置中指定的兼容版本匹配。

F C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\kernels\conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo(), &algorithms)

我的tensorflow版本为1.6，CUDA版本为：Cuda compilation tools，release 9.0，V9.0.176。

有人知道这里出了什么问题吗？

- user2505650

---

您安装的CUDNN版本不正确。 - talonmies

我刚刚下载了正确的驱动程序，但仍然出现错误。有没有办法检查我是否拥有正确版本的CUDNN？ - user2505650

是的，有的，而且你已经完成了。错误信息会准确告诉你安装的CuDNN版本以及所需版本。 - talonmies

我应该得到7.0吗？我在Nvidia的网站上找不到它。 - user2505650

Answer 1

您需要安装cuDNN 7.0.5。文件可以从此处下载。点击下载并同意条款后，选项将被列出。

- Rookie

---

我可以问一下如何找出需要哪个版本吗？（比如，你怎么知道他需要7.0.5）。我下载了cuda 9.0，然后又下载了cuda 9.0的cudnn 7.1.2，所以我觉得很奇怪为什么它们不兼容。 - AllBecomesGood

请注意，收到的错误信息如下：已加载运行时CuDNN库：7102（兼容性版本7100），但源代码编译时使用的是7003（兼容性版本7000）。这表明用户已安装cuDNN 7.1（7100）版本，但源代码库是使用cuDNN 7.0（7000）编译的，导致不兼容。7000系列的最新版本是cuDNN 7.0.5。 - Rookie