码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2016-09-01 09:17:31      阅读:5646      评论:0      收藏:0      [点我收藏+]

标签:

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

技术分享

技术分享

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

标签:

原文地址:http://www.cnblogs.com/mafang/p/5828330.html

(1)
(1)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!