(原)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 2

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

如果要使用多个GPU,可以使用如下命令:

./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 0,1,2,3

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

如果要使用所有的GPU,可使用如下命令:

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

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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM