Как использовать обучение с несколькими GPU с помощью Python с использованием Caffe (pycaffe)?

Question

Я видел, что недавно в Caffe был добавлен new commit, который должен включать обучение с несколькими GPU с пользовательскими слоями Python/Python. Я не могу тренироваться в сети с использованием нескольких GPU из моего кода Python. Я клонировал обновленный репозиторий и создал Caffe с USE_NCCL. команды я пытался использовать в моем сценарии Python:

caffe.set_mode_gpu() 
caffe.set_solver_count(2) 

Я получаю ошибку:

F0223 10: 27: 41.249380 23468 python_layer.hpp: 25] PythonLayer не поддержки CLI Мульти -GPU, используйте train.py

При запуске сценария из командной строки он просто использует только 1 графический процессор.

Answer 1

Я также недавно попробовал поддержку нескольких GPU в Python Caffe. Вы можете дать этому прочитанный: https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

Caffe поддерживает только мульти-GPU из командной строки, и только во время ПОЕЗДЕ т.е. вы должны использовать файл train.py (./build/tools/caffe поезд) и дать GPU, который вы хотите использовать в качестве аргументов для этого скрипта. Это довольно хорошо объяснено в файле, который я указал вам выше.

Я попытался использовать caffe.set_solver.count (2) в моем скрипте python. Я не получил никаких ошибок, однако моя модель тренируется только на одном графическом процессоре.

-Ayushi