转载请注明出处:

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

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

 

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

 

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

 

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

相关文章:

  • 2022-02-13
  • 2021-05-18
  • 2021-10-08
  • 2022-02-03
  • 2022-01-29
  • 2021-05-16
  • 2022-12-23
  • 2022-01-25
猜你喜欢
  • 2022-03-08
  • 2021-07-07
  • 2021-06-05
  • 2021-10-03
  • 2022-02-20
  • 2021-05-27
  • 2022-02-23
相关资源
相似解决方案