(原)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还是什么原因。。。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM