将Caffe与多个GPU一起使用错误:为int32标志'gpu'指定了非法值'0,1'

问题描述

我正在尝试使用带有两个GPU的Caffe训练CNN。 enter image description here './build/tools/caffe train --solver = examples / mnist / lenet_solver.prototxt -gpu 0,1'

然后出现错误错误:为int32标志'gpu'指定了非法值'0,1'

但是,我只能使用一个GPU来训练网络: “ ./build/tools/caffe train --solver = examples / mnist / lenet_solver.prototxt -gpu 0'”和 './build/tools/caffe train --solver = examples / mnist / lenet_solver.prototxt -gpu 1'

如果我使用'-gpu all',则错误:为int32标志'gpu'指定了非法值'all'

我也尝试过'-gpus',不起作用。 enter image description here

在Caffe中是否需要设置某些内容? Ang的建议非常受欢迎。

干杯, 毛

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)