1. 程式人生 > >Ubuntu14.04 安裝 Caffe

Ubuntu14.04 安裝 Caffe

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