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

用caffe训练mobilenet的时候总是出现如下错误
【caffe训练mobilenet】报错:Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ER_第1张图片

由于我是双显卡的配置,之前已经把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。

你可能感兴趣的:(【caffe训练mobilenet】报错:Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ER)