(原)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