zoukankan      html  css  js  c++  java
  • (原)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还是什么原因。。。

  • 相关阅读:
    [CERC2015]Digit Division
    [SDOI2012]拯救小云公主
    [不知道出自什么OJ]最大利润
    [USACO08FEB]修路Making the Grade
    P1985 [USACO07OPEN]翻转棋
    P1156 垃圾陷阱
    奶酪
    扫描线
    php数组
    php switch
  • 原文地址:https://www.cnblogs.com/darkknightzh/p/6230227.html
Copyright © 2011-2022 走看看