(原)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 阅读(14192) 评论(0) 编辑 收藏 举报