DeepLearning: CPUのみでCaffeを使うときの注意点
Ubuntu14.04にCaffeをインストール(CPUモード) - Qiitaを参考にして、VMware上のUbuntuにCaffeをインストールしました。
注意点1:
前述のサイトの通り、CUDAもインストールして、Caffeをインストール。問題無いように思えたが、Ubuntuを再起動するとGUIが起動しなくなった。。。
原因はCUDAをインストールしたことがどうか特定はできていないが、CUDAをインストールせずに、CaffeをCPUのみのモードでインストールしたところ、問題無くGUIも起動!
注意点2:
CaffeでTrainingを動かそうとすると、下記のエラーが発生。。。
$ XXXX/caffe/build/tools/caffe train --solver xxxxx.prototxt
I0131 22:24:17.640139 4868 caffe.cpp:185] Using GPUs 0
F0131 22:24:17.640458 4868 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.
*** Check failure stack trace: ***
@ 0x7fc6511f2daa (unknown)
@ 0x7fc6511f2ce4 (unknown)
@ 0x7fc6511f26e6 (unknown)
@ 0x7fc6511f5687 (unknown)
@ 0x7fc65158439c caffe::Caffe::SetDevice()
@ 0x409c69 train()
@ 0x405e2b main
@ 0x7fc6504ffec5 (unknown)
@ 0x406405 (unknown)
@ (nil) (unknown)
Aborted (core dumped)
xxxxx.prototxt でCPU modeかGPU modeか指定する必要がある。何も指定しないと同じエラーが出たので、恐らくデフォルトはGPU mode。
# GPU mode solver_mode: GPU
or
# CPU mode
solver_mode: CPU