Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

       这是我在跑caffe的例子mnist时出现的问题;
       遇到Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM的问题,数字可能变,主要原因是自己的GPU计算力不行,内存不足, 查看自己的cuda capability是不是小于3(查看cuda capability的方法是:sudo ./usr/local/cuda/samples/1_Utilities/deviceQuery/deviceQuery),这时候就不要用GPU了;
      我的机子是1080Ti,cuda capability是6.1,不可能是因为内存问题,我看见有人把nvida的驱动卸载了重装就好了,不过好像中间还有几个坑;

     我这里有更简单的方法,sudo gedit caffe/examples/mnist/lenet_train_test.prototxt,在convolution_param中添加engine:caffe

     然后问题就解决了。

     

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值