【caffe训练mobilenet】报错:Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ER

用caffe训练mobilenet的时候总是出现如下错误
这里写图片描述

由于我是双显卡的配置,之前已经把Makefile.config里的use_nccl去掉了注释重新编译了,但是实在是找不出问题。而且用两个显卡跑出现的错误不一样,一个是(4 vs.0),另一个是(1 vs.0)
因为模型是师兄给我的,所以问了师兄,他说把Makefile.config里的use_cudnn的注释加上重新编译就可以了,现在还是很迷惑,但总算跑起来了,速度挺慢,后续再增加内容吧,目前也不清楚原理。


2019.04.01
查看到的另外的解决办法是命令前加上sudo

若出现CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

修改CMakeLists.txt的第41行,即如下的行;

caffe_option(USE_CUDNN “Build Caffe with cuDNN library support” OFF IF NOT CPU_ONLY)
把ON该成OFF。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值