【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的注释加上重新编译就可以了,现在还是很迷惑,但总算跑起来了,速度挺慢,后续再增加内容吧,目前也不清楚原理。

猜你喜欢

转载自blog.csdn.net/qq_17130909/article/details/78678346