あるエンジニアのひとり旅

大企業を辞めたエンジニア、研究者のちょっとした日記

E-mail: miraimage.lab@gmail.com, Twitter: @miraimage_lab

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