Ubuntu14.04配置caffe

Ubuntu14.04配置caffe

轉載 2016年09月26日 16:41:47
  • 676
前段時間接到任務需要配置caffe並在上面訓練神經網絡相關,對caffe與深度學習一竅不通的我有幸得到了xizero00師兄的指導並在Ubuntu 14.04上面配置了caffe,親測可用,以下是師兄寫的博文正文:

網上充斥各種安裝方法,但是都很繁瑣,特別是安裝顯卡驅動以及依賴項,對於初學者不是很方便。我覺得采用包管理工具更方便因此寫下本文。


(1)首先安裝ssh,這樣能夠使用winscp和putty或者secure crt進行遠程連接,方便在windows下載東西傳到linux下
sudo apt-get install ssh

(2)安裝caffe的依賴項
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

2)BLAS依賴項
sudo apt-get install libatlas-base-dev
或者安裝
sudo apt-get install libopenblas-dev



3)python依賴項
安裝python及其頭文件
sudo apt-get install python
sudo apt-get install python-dev

安裝python的其他依賴
sudo apt-get install python-numpy
sudo apt-get install ipython
sudo apt-get install ipython-notebook
sudo apt-get install python-sklearn
sudo apt-get install python-skimage
sudo apt-get install python-protobuf


4)谷歌glog和gflags和lmdb依賴項
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

(3)然後安裝cuda和顯卡驅動
用迅雷下載
然後用winscp傳輸到linux下,運行如下命令
sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install cuda

1)安裝完成後設置環境變量
sudo gedit /etc/bash.bashrc
在文件的末尾加入如下
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

2)安裝完畢後驗證顯卡驅動是否安裝完畢(如果檢測不到或者提示找不到文件可以重啓下系統,因爲剛裝完顯卡驅動)
cat /proc/driver/nvidia/version

3)安裝完顯卡驅動之後可以安裝cuda的sample,<dir>就是你要安裝的位置
cuda-install-samples-7.5.sh <dir>

(4)安裝git,並且下載代碼
sudo apt-get install git
然後進入到源碼目錄
cd caffe

(5)編譯Caffe
cp Makefile.config.example Makefile.config
然後修改Makefile.config設置你的python的路徑和matlab的路徑


注意:-j8表示使用多核進行編譯,8是核心個數

make all -j8
make test -j8
make runtest
如果要編譯python的綁定(你要安裝好python及其依賴)
make pycaffe -j8
如果要編譯matlab的綁定(你要安裝好matlab)
make matcaffe -j8

附註:

cudnn這裏沒寫。也很簡單。在nvidia網站下載一個cudnn的deb包安裝一下就可以了 


enjoy it!


written by xizero00

http://blog.csdn.net/xizero00


以上是轉載內容

附原文地址:http://blog.csdn.net/xizero00/article/details/52549883

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