用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。