歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux基礎 >> Linux教程 >> Mac下快速安裝Caffe並訓練MNIST

Mac下快速安裝Caffe並訓練MNIST

日期:2017/2/28 13:49:14   编辑:Linux教程

在Ubuntu上成功安裝了Caffe,但是沒有訓練MNIST,今天在Mac上極簡安裝了Caffe並且順利地跑通了MNIST。網絡上的教程很多都比較復雜,這篇文章僅關注安裝和訓練過程,不涉及深度學習的理論。

希望看到這篇文章的朋友不要去關注GPU、cuCNN、CUDA、cmake、Python、Matlab、LeNet、libstdc++、OpenBLAS、MKL等名詞,這些只會讓人覺得混淆。

第一手的資料當然來自Caffe的官方網站,裡面的內容很多,很容易讓人困惑,以下我按照自己的實際情況來說明整個安裝過程。

Caffe安裝

首先按照我這裡的 http://www.linuxidc.com/Linux/2016-09/135027.htm 安裝Homebrew和OpenCV,我安裝的OpenCV版本是3.1.0。在終端輸入以下命令查看OpenCV的版本,

pkg-config --modversion opencv

此後安裝一些必要的庫,在終端中輸入以下命令,

sudo brew install -vd snappy leveldb gflags glog szip lmdb
sudo brew tap homebrew/science
sudo brew install hdf5

以上是Caffe提供的命令,我把其中的OpenCV的安裝命令去掉了,因為按照之前的文章單獨安裝了OpenCV。

此後安裝boost和protobuf,用最簡單的方式,如下,

sudo brew install protobuf boost

BLAS是Mac中自帶的庫,所以不需要安裝。至此Caffe所需的第三方庫就安裝完畢,很簡單,真的很簡單,只是網絡上的教程給人一種很復雜的感覺。

下面是Caffe本身的安裝過程,為了方便將Caffe克隆到桌面上,我下載的Caffe的路徑是’/Users/mac/Desktop/caffe’,用$CAFFE_ROOT來代替Caffe的主路徑。其中需要將Makefile.config.example拷貝一份成Makefile.config,這樣之後才能make。

另外Makefile.config需要進行修改,去掉CPU_ONLY := 1的注釋,這樣Caffe就只運行在CPU上了,另外對我來說需要去掉OPENCV_VERSION := 3的注釋,因為我的OpenCV版本是3.1.0,根據自己的情況選擇,如果OpenCV版本是2.*就不需要去掉注釋。

git clone https://github.com/BVLC/caffe
cd /Users/mac/Desktop/caffe
cp Makefile.config.example Makefile.config
make all
make test
make runtest

在make的過程中會看到很多warnings,不需要在意,如果看到error說明安裝有錯誤。在最後一步runtest後可以看到如下結果,表示Caffe安裝成功,通過了測試。

MNIST訓練

之前將Caffe的環境搭好了,現在用MNIST這個數據集進行測試,繼續在$CAFFE_ROOT下進行操作,首先Mac需要安裝wget才能執行以下.sh的腳本。

sudo brew update(如果需要的話)
sudo brew install wget
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

經過上述操作./examples/mnist/路徑下會有mnist_test_lmdb和mnist_train_lmdb兩個文件夾,分別是測試和訓練數據。

在最終訓練之前需要修改./examples/mnist/lenet_solver.prototxt最後一句話為

solver_mode: CPU

這樣保證整個訓練過程在CPU上進行,最後執行訓練命令,

cd $CAFFE_ROOT
./examples/mnist/train_lenet.sh

看到如下效果,

一共迭代10000次,准確率為0.9899,最後訓練的model為./examples/mnist/lenet_iter_10000.caffemodel。

以上就是整個Caffe安裝和訓練MNIST的過程,安裝過程可能會遇到很多問題是我沒有提到的,但是要保持一個想法就是這個過程不麻煩,這是很關鍵的。

Ubuntu 16.04下Matlab2014a+Anaconda2+OpenCV3.1+Caffe安裝 http://www.linuxidc.com/Linux/2016-07/132860.htm

Ubuntu 16.04系統下CUDA7.5配置Caffe教程 http://www.linuxidc.com/Linux/2016-07/132859.htm

Caffe在Ubuntu 14.04 64bit 下的安裝 http://www.linuxidc.com/Linux/2015-07/120449.htm

深度學習框架Caffe在Ubuntu下編譯安裝 http://www.linuxidc.com/Linux/2016-07/133225.htm

Caffe + Ubuntu 14.04 64bit + CUDA 6.5 配置說明 http://www.linuxidc.com/Linux/2015-04/116444.htm

Ubuntu 16.04上安裝Caffe http://www.linuxidc.com/Linux/2016-08/134585.htm

Caffe配置簡明教程 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) http://www.linuxidc.com/Linux/2016-09/135016.htm

Ubuntu 16.04安裝Caffe並訓練MNIST http://www.linuxidc.com/Linux/2016-09/135025.htm

Copyright © Linux教程網 All Rights Reserved