2016-12-13 24 views
0

Makefile.config.exampleをコピーして編集せずにCaffeをコンパイルしました。私はcaffeでコードを実行しようとし、それは私にCPUモードを示しています。私の質問は、Caffeのインストール後に設定ファイルを洗練することが可能かどうかです。はいの場合、どうですか?caffeでGPUモードとCPUモードを切り替える方法は?

~/caffe-master$ SOME_COMMANDS BLAH BLAH 

CPU mode 

私はGPUモードおよびその逆にCPUモードから実行している切り替えることができますどのように?

本当にありがとうございます。私はcaffeとpythonの両方で初心者です。ありがとう

+0

GPUをサポートしていますか?デフォルトで編集していない場合は、CPUのみのビルドが行われます。 –

+0

@LukaszTracewski私は設定ファイルを変更しませんでした。これらの変更をどのようにしてCPUからGPUに行うことができますか?私はcaffeを取り除き、最初からそれをコンパイルするべきですか? –

+0

Windowsやubuntuで 'caffe'をどうやってコンパイルしましたか? – Dale

答えて

0

あなたはどこであなたのCaffeを手に入れていますか? official BVLC releaseはデフォルトでGPUビルドとして提供されます。

# CPU-only switch (uncomment to build without GPU support). 
# CPU_ONLY := 1 

あなたはCPUのみの構築取得する​​には、こののコメントを解除する必要があります。

まず、Makefile.configは線対を持っています。

次に、各ソルバーファイルの最後の行は、あなたがCPUの実行を取得するためにCPUにこれを変更する必要が

solver_mode: GPU 

です。

+0

あなたの返事をありがとう。私は 'Makefile.config'をチェックしました、それは既にコメントされています。そして 'solver_modeはGPUにあります。 ' –

+0

これは私にとっては変です。それらを変更することを忘れてしまった場合、ソフトウェアは私にGPUモードで実行しようと叫んでいる。 – Prune

0

まず、 "Makefile.config.example"を編集する必要はありません。 " \ windows \ CommonSettings.props.example"です。すべての前に

だけのような何か、あなたのコマンドの末尾に「-gpuすべて」を使用してカフェを実行します。その後、

>caffe train lenet_solver.prototxt -gpu all 

それが再びCPUで行った場合、私はあなたを再構築する必要があると思います上記のファイルに必要なエディションのcaffe。もちろん、新しいバージョンのcaffeでは、 ".props"ファイルのデフォルトモードでGPUオプションがありますが、CUDA_PATHは次のようにシステム変数(環境変数)に設定する必要があります:

CUDA_PATH = C:\プログラムファイル\ NVIDIA GPUコンピューティングツールキット\ CUDA \ v7.5

あなたのcudaが正しくインストールされている場合、自動的に設定する必要があります。

関連する問題