在Ubuntu 14.04上安裝Caffe

安裝依賴

sudo apt-get install libatlas-base-dev
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

安裝Caffe

git clone https://github.com/BVLC/caffe.git
cd caffe
cp Makefile.config.example Makefile.config

我是在服務器上,選擇只用CPU(不用GPU)計算

vi Makefile.config
# Adjust Makefile.config (for example, if using Anaconda Python)
# uncomment CPU_ONLY := 1

最後編譯

make all
make pycaffe

跑一下測試(可選)

make test
make runtest

使用Caffe

跑MNIST試試

cd caffe
sh data/mnist/get_mnist.sh
sh examples/mnist/create_mnist.sh
vi examples/mnist/lenet_solver.prototxt
# 修改 solver_mode 爲 CPU
./examples/mnist/train_lenet.sh

更多教程我在看

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