机器有多个gpu,如何指定特定gpu参与模型训练和预测。
Created by: yinfeigl
类似caffe可以通过指定-gpu选项(如下),实现指定gpu运行,paddle如何完成设置呢? ./build/tools/caffe train --solver=examples/testXXX/solver.prototxt # 使用默认的gpu0 ./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 2 ./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu 0,1,2 ./build/tools/caffe train --solver=examples/testXXX/solver.prototxt --gpu all