ubuntu安装caffe

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler -y

sudo apt-get install --no-install-recommends libboost-all-dev -y
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev -y
sudo apt-get install libatlas-base-dev -y 
sudo apt-get install python-dev -y
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags  ipython -y

下载caffe源码

git clone https://github.com/BVLC/caffe.git

修改编译配置文件:

cd caffe

复制示例文件为编译配置文件:

cp Makefile.config.example Makefile.config​ 

修改Makefile.config​ ​ 文件

nano Makefile.config

GPU版:将#USE_CUDNN := 1 修改成: USE_CUDNN := 1

CPU版(无显卡)将#CPU_ONLY := 1 修改成  CPU_ONLY := 1

 

将# Whatever else you find you need goes here.下面的

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib

修改为:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

 

 

打开makefile文件,做如下修改:

将:NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)

替换为:

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

 

 

make all -j20                                    

make test -j20

make runtest -j20

make pycaffe 

 make错误的时候可以   make clean 清理一下,重新编译

echo export CAFFE_ROOT=/home/$USER/caffe >>~/.bashrc
echo export PYTHONPATH=/home/$USER/caffe/python:$PYTHONPATH >> ~/.bashrc
source ~/.bashrc

 

 

mnist测试

 

cd caffe

sh data/mnist/get_mnist.sh

sh examples/mnist/create_mnist.sh

vi examples/mnist/lenet_solver.prototxt  #最后一行修改模式CPU or GPU

 

time sh examples/mnist/train_lenet.sh

 

 

cannot find -lboost_python3 错误:  

lboost_python3        cd /usr/lib/x86_64-linux-gnu/                   sudo ln -s libboost_python-py35.so libboost_python3.so

 

No module named skimage.io     --------     pip install scikit-image

 

 

fatal error: numpy/arrayobject.h   :     

import numpy as np  
np.get_include()  
得到:

/usr/local/lib/python2.7/dist-packages/numpy/core/include

在Makefile.config找到PYTHON_INCLUDE,

 

 

要加一个local,变成: 

PYTHON_INCLUDE := /usr/include/python2.7 \
        /usr/local/lib/python2.7/dist-packages/numpy/core/include
再make pycaffe就ok了

 

 

 

 ValueError, "Can't create weekday with n == 0"  

   sudo pip3 install python-dateutil --upgrade

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值