(原)torch使用caffe时,提示CUDNN_STATUS_EXECUTION_FAILED

转载请注明出处:

http://www.cnblogs.com/darkknightzh/p/6230227.html

 

提前说明:此文不能真正解决该问题,具体原因我也不知道。。。

 

以前使用某台电脑A上torch来require “caffe”后,使用net:forward时,能正常运行,这两天换了另一台电脑B,运行程序,就出现下面的错误:

Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0)  CUDNN_STATUS_EXECUTION_FAILED

 

 

如果不使用cuda速度太慢,使用又出错。后来仔细查找原因,发现是A编译caffe时,并没有编译cudnn,所以程序只使用cuda了。电脑B上面编译了cudnn,但是路径设置的不对,导致torch中调用时,出现了上面的错误。

 

改正方法:重新编译caffe,取消使用cudnn,此时torch中的程序能正常运行。暂时只能这样了。。。(修改B电脑cudnn路径并重新编译caffe后,torch中还是无法使用cudnn,所以只能暂时不用了。。。)

 

说明:其实还有一点不明白的,在caffe中直接训练时,或者在python中使用caffe时,并没有出错,不清楚是没有使用cudnn还是什么原因。。。

posted on 2016-12-28 17:16  darkknightzh  阅读(14176)  评论(0编辑  收藏  举报

导航