Caffeが複数のGPUを使用しているかどうか確認する

Question

トレーニングニューラルネットワークを初めて使用しています。私はGPUクラスタにアクセスすることができ、シーン分類のためにAlex-Netのバージョンを微調整しています。

私は現在2つのGPUにアクセスしています。私はトレーニングのためにその両方を使いたいと思います。 nvidia-smiコマンドは私にGPUのID（0と1）を与えます。

これは私がGPUの両方を使用する訓練を行う方法です：（[0,1]）

caffe.set_mode_gpu（）
caffe.set_deviceが、これはそれを使用する正しい方法ですか？

Answer 1

Pythonでは、set_device()を使用して1つのGPUを選択できます。マルチGPUは、C++インターフェイスでのみサポートされています。この目的のために使用される--gpuフラグについては、hereを参照してください。トレーニングに使用するGPUは、Caffeツールのコマンドラインで--gpuフラグを使用して設定できます。例えば、

build/tools/caffe train --solver=models/bvlc_alexnet/solver.prototxt --gpu=0,1 

は、あなたがそのように入力することができ、端末でのGPU 0と1

Answer 2

に訓練します：

python yourpythonfile.py CUDA_VISIBLE_DEVICES=0, 1 

を