Ubuntu 14.04 下 caffe 的安装

Ubuntu 14.04 下 caffe 的安装

官方安装指导网站:

http://caffe.berkeleyvision.org/installation.html

1、General dependencies

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

2、CUDA安装

NVIDIA公司官网下载cuda最新版本runfile安装包,下载完以后使用md5sum校验下载文件是否正确,网址:

https://developer.nvidia.com/cuda-downloads

顺便把pdf的安装guide文件下载下来。

按照文档

Chapter 2. Pre-installation Actions

Chapter 6 Pre-installation actions

Chapter 4. Runfile Installation

顺序安装


3、BLAS: install ATLAS by

sudo apt-get install libatlas-base-dev

or install OpenBLAS or MKL for better CPU performance.


4、Python:系统默认自带Python2.7,但还需要安装python-dev

sudo apt-get install python-dev


5、其余的安装

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

以上安装完,下载caffe工程,就能够编译通过

cp Makefile.config.example Makefile.config
# Adjust Makefile.config (for example, if using Anaconda Python)
make all
make test
make runtest

可以采用-j8,加速编译如:

make all -j8

如果需要安装pycaffe,还需要安装python目录下,

requirements.txt
文件中列举的软件,root权限下使用以下命令安装(在python目录下执行)

for req in $(cat requirements.txt); do pip install $req; done

安装中会出错,提示缺少fortran编译器,需要先安装编译器
sudo apt-get install gfortran
安装完以后列表中软件就能全部安装成功。

6、安装pycaffe

在caffe工程根目录下,安装pycaffe

make pycaffe

可能会提示找不到numpy里面的头文件,这是因为安装numpy的目录和makefile.config文件里指定的目录不一样。默认目录为

PYTHON_INCLUDE := /usr/include/python2.7 \
		/usr/lib/python2.7/dist-packages/numpy/core/include
但numpy可能在

PYTHON_INCLUDE := /usr/include/python2.7 \
		/usr/local/lib/python2.7/dist-packages/numpy/core/include
修改makefile.config文件就能够成功安装pycaffe。

7*、draw_net.py依赖软件

python功能中,draw_net.py可以可视化网络结构,需要使用,要先安装pydot和graphviz两个软件

pip install pydot
apt-get install graphviz

over!









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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值