如何在ubuntu下搭建深度學習caffe環境

        在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時,遇到的各種問題及解決辦法,可以作爲參考。

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章