Caffe（pycaffe）を使ったPythonでのマルチGPUトレーニングの使い方は？

Question

最近、new commitがCaffeに挿入されたことがわかりました。これは、Pythonインターフェイス/ PythonカスタムレイヤーでマルチGPUトレーニングを有効にする必要があります。 Pythonコードから複数のGPUを使ってネットワークを鍛えることができません。 更新されたリポジトリをクローンし、USE_NCCLを使用してCaffeをビルドしました。

caffe.set_mode_gpu() 
caffe.set_solver_count(2) 

エラー私が手：：

F0223 10：27：41.249380 23468 python_layer.hpp：25] PythonLayerはないの サポートCLIマルチ コマンドは、私は私のPythonスクリプトで使用してみました-GPU、train.pyを使用

スクリプトをコマンドラインから実行すると、単に1つのGPUだけが使用されます。

Answer 1

私も最近、Python CaffeでマルチGPUサポートを試みました。 https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

Caffeは、コマンドラインからのマルチGPUのみをサポートしています。つまり、train.py（./build/tools/caffe train）を使用する必要があります。このスクリプトの引数として使用したいGPUです。 私が上で指摘したファイルでは説明がうまく説明されています。

私のpythonスクリプトでもcaffe.set_solver.count（2）を使用しようとしました。しかし私のモデルは1つのGPUだけで列車に乗っていてもエラーは出ませんでした。

-yyushi