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

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,这个错误就没有啦~~当然我的又报了别的错....

posted @ 2016-09-01 09:03  同销万古愁  阅读(8425)  评论(0编辑  收藏  举报