ロボット研究者の戯言

caffeでcudnnのエラーが出た(CUDNN_STATUS_BAD_PARAM)

      2017/07/03

Caffeでこんなエラーが出た

F0302 10:07:33.978206 20193 cudnn.hpp:96] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

CuDNNがうまく動かないみたい…

考えられる原因は,

  1. CUDAのパスがおかしい
  2. 実行しているプログラムがおかしい

ただし,Caffeのビルド自体はできているので,1の可能性は低い.

2に関しては,カーネルサイズを変更するとエラーが出なくなるという報告もあるみたいだけど,実行しようとしているプログラムは誰かが作ったサンプルコードで,ちゃんと理解できていないのでできるだけ変更したくない.

さて,どうしよう.

スポンサーリンク



結局どうしたか

CaffeのMakefile.configを変更して,CuDNN使わないようにしてリビルドしました.

もちろん実行速度はCuDNNを使った方が速いでしょうが,サンプルを試してみたかっただけなので,これでとりあえずOKとしました.

根本的な解決にはなっていないのですが...

 - Caffe, Ubuntu, 機械学習 , , , , , ,