Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal

时间:2016-09-01 09:17:31   收藏:0   阅读:5646

ubuntu14.04 caffe 运行别人的程序,报错Check failed: error == cudaSuccess (10 vs. 0)  invalid device ordinal。根据网上的说法,GPU内存不够,可是batch大小改为1了还是不行。问了博士师兄才发现,在该行上面有一行,写着using GPUs 1.而我只有一个GPU,如图2,我把solver_48.prototxt中最后一行的device_id改为0,接着运行,见图1,这个错误就没有啦~~当然我的又报了别的错....

技术分享

技术分享

评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!