機器學習 - Ubuntu14.04 x64 搭建 Caffe

寫一下如何在 Ubuntu 14.04上搭建 Caffe

CPU模式

這個就是不使用 GPU 加速的，暫時我現在也只使用了這個方式，後面等我買的主機到了，再寫 GPU 模式的。

1. 下載必備依賴

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

拷貝到終端裏執行，這是一些 Ubuntu 安裝 caffe的必備的依賴。

1. 下載 BLAS

sudo apt-get install libatlas-base-dev //一個基礎線性代數數學庫

1. 下載 Python

sudo apt-get install python-dev

一般這個系統自帶，不帶就安裝下。

1. 繼續下載一些依賴

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

繼續輸入執行。

1. 下載 caffe

git clone https://github.com/BVLC/caffe

然後進入到這個文件加里

cd caffe/
cp Makefile.config.example Makefile.config

然後編輯 Makefile.config這個文件，因爲我們沒有安裝 Cuda所以把第8行的 CPU_ONLY=1的註釋取消掉，意思是隻使用 CPU.

1. 編譯並測試

make all
make test
make runtest

如果覺得慢在編譯命令後面加-j就是：make all -j這種，如果是多核會加速的。

如果沒報錯，就可以使用了。

然後可以用樣例跑一下試試：

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
vi ./examples/mnist/lenet_solver.prototxt

把裏面的solver_mode改爲CPU然後執行訓練腳本：

謹記一定要在 caffe的根目錄下運行，不能cd examples/mnist然後執行./create_mnist.sh這樣子操作，會失敗的！！

./examples/mnist/train_lenet.sh

這樣就 OK 了！
運行圖示：