（原）caffe在ubuntu中设置GPU的ID号及使用多个GPU

转载请注明出处：

http://www.cnblogs.com/darkknightzh/p/5884892.html

参考网址：

http://caffe.berkeleyvision.org/tutorial/interfaces.html

由于在ubuntu中使用caffe的程序时，都使用.sh文件，该文件中常见的命令为：

./build/tools/caffe train --solver=examples/testXXX/solver.prototxt

当电脑中有多个GPU时，默认使用GPU0。如果想使用其他的GPU，可以将该文件内容修改如下：

./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 

注意，caffe中默认编号从0开始，因而--GPU 2的意思是使用第3个GPU。

如果要使用多个GPU，可以使用如下命令：

./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu ,,,

则使用0,1,2,3这4个GPU。

如果要使用所有的GPU，可使用如下命令：

./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu all

注意，使用的GPU越多，开始初始化时时间越久。当然，训练速度越快。