首页 > 代码库 > Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal
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,这个错误就没有啦~~当然我的又报了别的错....
Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。