轉載請注明出處:
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還是什么原因。。。