2021-05-12 14:32:11
Ubuntu 16.04安裝 Caffe GPU版
2020-06-16 17:06:48
安裝環境：Ubuntu 16.04 lts 64位元, gcc5.4 gpu1080ti，cuda8.0，cudnn5.1.10
1. 安裝依賴庫
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev libboost-all-dev protobuf-compiler sudo apt-get install libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev
2. 安裝caffe
終端輸入命令：
git clone git://github.com/BVLC/caffe.git
然後找到caffe資料夾，開啟後會發現Makefile.config.example 檔案，
這時候將Makefile.config.example複製一份命名為Makefile.config，並開啟，將
#USE_CUDNN := 1 # CUDA_DIR := /usr/local/cuda # CUDA_ARCH := -gencode arch=compute_20,code=sm_20 #-gencode arch=compute_20,code=sm_21 #-gencode arch=compute_30,code=sm_30 #-gencode arch=compute_35,code=sm_35 #-gencode arch=compute_50,code=sm_50 #-gencode arch=compute_50,code=compute_50
去掉註釋#，並且CUDA_DIR 更換為你的cuda安裝路徑
開啟Makefile.config檔案，將
/usr/include/hdf5/serial
新增到
INCLUDE_DIRS := $(Python_INCLUDE)/usr/local/include
開啟Makefile檔案，找到
LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
更改為：
LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
之後執行
make all make test make runtest
若在編譯過程中報錯出現丟失檔案的情況，可以尋找丟失檔案路徑。
例如常見問題：
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
解決方法：找到檔案路徑：
sudo find / -name hdf5.h
找到對應檔案路徑，將其加入Makefile.config中。
查詢結果為：
/usr/include/hdf5/serial/hdf5.h
開啟Makefile.config，在
INCLUDE_DIRS := $(PYTHON_INCLUDE)/usr/local/include
之後新增
/usr/include/hdf5/serial
注意路徑，有INCLUDE_DIRS 和 LIBRARY_DIRS兩個，看清丟失檔案的路徑在/usr/include中還是/usr/lib中
正常情況下應該是沒問題了。
3. 安裝anaconda，設定python caffe介面
1.安裝anaconda ，選擇Python2.7 注意檔名
bash Anaconda2-4.3.0-Linux-x86_64.sh
2.安裝過程中注意提示，千萬不要一直enter，記得一直選擇預設的就行，傻瓜式安裝，之後會提示是否在bashrc文件中新增路徑，這裡選擇“yes”
3.開啟bashrc文件，在Home下，開啟顯示隱藏檔案選項，就可以找到該檔案。在最後一行裡加入
export PATH=/home/（你的使用者名稱）/anaconda2/bin:$PATH（具體路徑根據你的安裝路徑而定） export PYTHONPATH=/home/（你的使用者名稱）/caffe/python:$PYTHONPATH（同上） export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6
Bashrc文件更改後需要source一下，或者登出/重新啟動電腦：
4. .修改caffe下的Makefile.config文件
#ANACONDA_HOME := $(HOME)/anaconda2 #PYTHON_INCLUDE := $(ANACONDA_HOME)/include #$(ANACONDA_HOME)/include/python2.7 #$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include #PYTHON_LIB := $(ANACONDA_HOME)/lib #WITH_PYTHON_LAYER := 1
去掉註釋#，將
PYTHON_LIB := /usr/lib
加上註釋。
5. 升級Anaconda中的gcc版本
conda update --all
或者
conda install libgcc
出現升級確認提示，輸入y，升級成功後，關閉終端重新開啟
6. 回到caffe目錄下，編譯
make all make pycaffe
成功後，輸入python，接著輸入
import caffe
出現錯誤提示：
No module named google.protobuf
此時輸入命令
conda install protobuf
結束後再重新import caffe，成功的話應該沒有任何提示。
4. 測試
進入caffe安裝目錄
./data/mnist/get_mnist.sh ./examples/mnist/create_mnist.sh ./examples/mnist/train_lenet.sh
成功的話那麼久就可以結束安裝了！
Ubuntu 15.04 下Caffe + + CUDA 7.0 安裝設定指南 http://www.linuxidc.com/Linux/2016-11/137497.htm
Caffe 深度學習入門教學 http://www.linuxidc.com/Linux/2016-11/136774.htm
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(CPU only) http://www.linuxidc.com/Linux/2016-09/135034.htm
本文永久更新連結地址：http://www.linuxidc.com/Linux/2017-09/147111.htm
相關文章