在ubuntu下搭建caffe環境是比較困難的,會遇到各種坑,當時不用害怕,遇到問題解決問題就行了。
1、安裝git,命令如下
sudo apt-get install git
2、安裝caffe依賴的庫
需要安裝的庫比較多,如下所示
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
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
3、下載caffe源碼
首先在計算中計劃好caffe的安裝位置,我的計算機中安裝的目錄爲
/home/bigmarshal/Documents/deep_learning/
進入到這個目錄,在終端上執行下面的命令
git clone https://github.com/bvlc/caffe.git
成功執行完這一步後,源碼就已經下載到計算機中了,在deep_learning文件夾下生成了一個caffe文件夾,就是caffe的源碼。
4、配置編譯參數
這一步很簡單,caffe的源碼中已經帶了一個配置文件,我們進入到caffe文件夾,執行下面的命令
cp Makefile.config.example Makefile.config
這個名字的作用是將示例的配置文件複製爲使用的配置文件名字。
用gedit打開Makefile.config文件,在這個文件中,我們基本不用修改,在文件的上邊有一行,如下所示
# CPU_ONLY := 1
這句的配置是配置是否只使用CPU模式,如果只使用CPU模式,則把前面的#號去掉,即解除這句代碼的註釋,初學者建議從CPU模式學起,因爲GPU模式的環境搭建過程會有更多的坑。
修改完之後,保存退出。
5、編譯caffe
編譯很簡單,執行下面的命令。
make -j8
至此,我們完成了caffe的安裝,看起來還比較簡單,但實際上,按照這個步驟安裝一次成功的概率很小,總會遇到各種坑,我的博客裏記錄了我在安裝caffe時,遇到的各種問題及解決辦法,可以作爲參考。