美文网首页
caffe 支持多gpu设置

caffe 支持多gpu设置

作者: sweetsky0901 | 来源:发表于2017-11-24 14:56 被阅读0次


正常训练的命令是

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

指定某块gpu卡工作:(如果不指定,是默认第一块在工作,指定3是第四块在工作)

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt --gpu x

指定多块卡一起工作(前提是编译caffe的时候,需要打开USE_NCCL)

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

相关文章

网友评论

      本文标题:caffe 支持多gpu设置

      本文链接:https://www.haomeiwen.com/subject/tiqfbxtx.html