Ubuntu14.04 安裝 Caffe
阿新 • • 發佈:2019-02-02
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