您的位置:首页 > 大数据 > 人工智能

Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

2018-03-22 20:51 1171 查看
       这是我在跑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
     然后问题就解决了。
     
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐