Caffe/pyCaffe: установить все графические процессоры

Question

можно установить все графические процессоры для Caffe (особенно pyCaffe)?

Что-то вроде:

caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all 

Answer 1

AFAIK Caffe не поддерживает мульти обучения GPU на данный момент. Это планируется для будущего выпуска. См. Обсуждение here.
Кажется, что NVIDIA's branch of caffe имеет эту функциональность. См. Вопрос here.

Answer 2

Кажется, что Caffe теперь начинает поддерживать обучение на нескольких GPU для интерфейса C++. См. Документы на CommandLine interface.

# train on GPUs 0 & 1 (doubling the batch size) 
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 0,1 
# train on all GPUs (multiplying batch size by number of devices) 
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all 

Answer 3

Обе вилки поддерживали мульти-графический процессор на некоторое время.

• BVLC/кофейная получил поддержку мульти-GPU на 08/13/2015 (см commit, issue).
• NVIDIA/caffe получила поддержку для нескольких GPU на 06/19/2015 (см. release note).

Вы можете быть заинтересованы, чтобы знать, что есть довольно серьезная нерешенный вопрос с несколькими GPU и PythonLayers (см issue, temporary fix).