1、安裝依賴包
sudo apt-get install build-essential git vim cmake libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler
2、安裝CUDA
(1)下載cuda倉庫安裝包
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb
(2)安裝倉庫
sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
(3)更新源
sudo apt-get update
(4)安裝CUDA及驅動
sudo apt-get install cuda
(5)配置環境變量
a.編輯/etc/profile文件:
sudo vim /etc/profile
b.在profile文件底部加入:
#cuda
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
c.使profile文件生效:
source /etc/profile
3、安裝ATLAS
sudo apt-get install libatlas-base-dev
4、獲取Caffe
git clone https://github.com/BVLC/caffe.git
5、編譯Caffe
cd caffe
cp Makefile.config.example Makefile.config
make all
make test
make runtest