kerasでのcnnの実行時のエラー

Question

現在、以下の環境でkerasを動かすことを考えております。

• Ubuntu16.04
• GPU GeForce 1070Ti
• cuda 9.0
• cudnn v7.1.1 Library for Linux
• Python3.5(Anaconda)
• tensorflow 1.6.0
• keras 2.1.4

基本的な環境構築は、

• https://qiita.com/mitsuharu_e/items/ecff16bdbcb795a86433
• https://www.tensorflow.org/install/install_linux#InstallingAnaconda

を元に行いました。

ところが、いざkerasのサンプルコードを走らせてみると、エラーが出てしまいます。

実行したコードはhttps://github.com/keras-team/keras/blob/master/examples/mnist_cnn.py
であり、

出力されたエラーメッセージは以下のとおりです。

2018-03-04 15:35:04.815615: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2018-03-04 15:35:04.932029: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:898] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
2018-03-04 15:35:04.932249: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1212] Found device 0 with properties: 
name: GeForce GTX 1070 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.683
pciBusID: 0000:01:00.0
totalMemory: 7.93GiB freeMemory: 7.60GiB
2018-03-04 15:35:04.932262: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1312] Adding visible gpu devices: 0
2018-03-04 15:35:05.792784: I tensorflow/core/common_runtime/gpu/gpu_device.cc:993] Creating TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 7344 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1070 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
2018-03-04 15:35:07.059592: E tensorflow/stream_executor/cuda/cuda_dnn.cc:378] Loaded runtime CuDNN library: 7101 (compatibility version 7100) but source was compiled with 7004 (compatibility version 7000).  If using a binary install, upgrade your CuDNN library to match.  If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
2018-03-04 15:35:07.060155: F tensorflow/core/kernels/conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms) 
中止(コアダンプ)

何が問題で、どのようにすれば解決できるのか、どなたかご教授いただけないでしょうか。

よろしくお願いいたします。

Answer 1

自己解決いたしました。
https://developer.nvidia.com/rdp/cudnn-download

こちらのCuDNNインストール画面で、
当初は

cuDNN v7.1.1 Library for Linux

を選んでおりましたが、こちらをアンインストールして、

cuDNN v7.1.1 Runtime Library for Ubuntu16.04
cuDNN v7.1.1 Developer Library for Ubuntu16.04
cuDNN v7.1.1 Code Samples and User Guide for Ubuntu16.04

からcuDNNをインストールしたら、無事kerasのコードが動きました。

Answer 2

Loaded runtime CuDNN library: 7101 (compatibility version 7100) but source was compiled with 7004 (compatibility version 7000).

CuDNNのバージョンを合わせてもうまく行きませんか？

Comments

[AnchorBlues]

mkgreiさん

ご返答ありがとうございます。

> CuDNNのバージョンを合わせてもうまく行きませんか？

ここでおっしゃっているCuDNNのバージョンというのは、上記に書いております
v7.1.1
のことでしょうか?
それとも、エラーメッセージにあります、
CuDNN library
のバージョンのことでしょうか?

参考までに申し上げておきますと、cudnnは、
https://developer.nvidia.com/rdp/cudnn-download
上記のページにおいて、
Download cuDNN v7.1.1 (Feb 28, 2018), for CUDA 9.0
から
cuDNN v7.1.1 Library for Linux
を選択してインストールしました。
CUDAのバージョンと対応したバージョンのCuDNNをインストールしたものと認識していたのですが、そうではなかったのでしょうか。