这是我在跑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
然后问题就解决了。