正常训练的命令是
./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
网友评论