【Caffe】Ubuntu 安装 Caffe gpu版

安装环境:Ubuntu 16.04lts 64位, gcc5.4 gpu1080ti,cuda8.0,cudnn5.1.10

1. 安装依赖库

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

2. 安装caffe

  终端输入命令:

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

  然后找到caffe文件夹,打开后会发现Makefile.config.example 文件,

  这时候将Makefile.config.example复制一份命名为Makefile.config,并打开,将

#USE_CUDNN := 1
# CUDA_DIR := /usr/local/cuda
# CUDA_ARCH := -gencode arch=compute_20,code=sm_20   #-gencode arch=compute_20,code=sm_21   #-gencode arch=compute_30,code=sm_30   #-gencode arch=compute_35,code=sm_35   #-gencode arch=compute_50,code=sm_50   #-gencode arch=compute_50,code=compute_50

  去掉注释#,并且CUDA_DIR 更换为你的cuda安装路径

  打开Makefile.config文件,将

/usr/include/hdf5/serial 

  添加到

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

  打开Makefile文件,找到

LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5

  更改为:

LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

  之后执行

make all
make test
make runtest

  若在编译过程中报错出现丢失文件的情况,可以寻找丢失文件路径。

  例如常见问题:

./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory

  解决方法:找到文件路径:

sudo find / -name hdf5.h

  找到对应文件路径,将其加入Makefile.config中。

  查询结果为:

/usr/include/hdf5/serial/hdf5.h

  打开Makefile.config,在

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

  之后添加

/usr/include/hdf5/serial 

  注意路径,有INCLUDE_DIRS 和 LIBRARY_DIRS两个,看清丢失文件的路径在/usr/include中还是/usr/lib中

  正常情况下应该是没问题了。

3. 安装anaconda,配置python caffe接口

1.安装anaconda ,选择Python2.7  注意文件名

bash Anaconda2-4.3.0-Linux-x86_64.sh

2.安装过程中注意提示,千万不要一直enter,记得一直选择默认的就行,傻瓜式安装,之后会提示是否在bashrc文档中添加路径,这里选择“yes”

3.打开bashrc文档,在Home下,打开显示隐藏文件选项,就可以找到该文件。在最后一行里加入

export PATH=/home/(你的用户名)/anaconda2/bin:$PATH(具体路径根据你的安装路径而定)
export PYTHONPATH=/home/(你的用户名)/caffe/python:$PYTHONPATH(同上)
export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6

  Bashrc文档更改后需要source一下,或者注销/重启电脑:

4. .修改caffe下的Makefile.config文档

#ANACONDA_HOME := $(HOME)/anaconda2

#PYTHON_INCLUDE := $(ANACONDA_HOME)/include   #$(ANACONDA_HOME)/include/python2.7   #$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include

#PYTHON_LIB := $(ANACONDA_HOME)/lib
#WITH_PYTHON_LAYER := 1

 去掉注释#,将

PYTHON_LIB := /usr/lib

 加上注释。

5. 升级Anaconda中的gcc版本

conda update

  或者

conda  install libgcc

  出现升级确认提示,输入y,升级成功后,关闭终端重新打开

6. 回到caffe目录下,编译

make all
make pycaffe

  成功后,输入python,接着输入

import caffe

  出现错误提示:

No module named google.protobuf 

  此时输入命令

conda install protobuf

  结束后再重新import caffe,成功的话应该没有任何提示。

4. 测试

  进入caffe安装目录

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
./examples/mnist/train_lenet.sh

  成功的话那么久就可以结束安装了!

时间: 2024-11-05 12:23:19

【Caffe】Ubuntu 安装 Caffe gpu版的相关文章

linux mint & ubuntu 安装QQ国际版

ubuntu安装QQ相对简单 下载qq国际版deb文件,直接安装即可. 下载地址: http://pan.baidu.com/s/1dDhAnTZ 下面就重点说一下linux mint 安装qq. 1.先将自己的source文件备份 sudo cp /etc/apt/sources.list /etc/apt/sources.list_backup 2.修改自己的source文件 deb http://mirrors.oschina.net/ubuntu/ raring main restric

ubuntu 安装 caffe 解决://home/xiaojie/anaconda/lib/libpng16.so.16:对‘[email protected]_1.2.9’未定义的引用

1. 当运行命令"make runtest -j8" 时出现上述问题,有两种解决方案: 1)GitHub上的解决方案,链接:https://github.com/BVLC/caffe/issues/6139 可以看出,是可以解决问题的!!! 2)执行命令: 1 git clone https://github.com/madler/zlib 2 cd path/to/zlib #这里黄色标注的换成自己的路径即可 3 ./configure 4 make 5 make install #

ubuntu安装caffe 解决:build_release/tools/caffe: error while loading shared libraries: libcudart.so.8.0: cannot open shar

1. 网上有3种解决方式: 1)这是GitHub的解决方法,链接:https://github.com/tensorflow/tensorflow/issues/5343 可以看出,这个方法还是有效的. 2)执行命令: 1 sudo ldconfig /usr/local/cuda/lib64 这个感觉也还可以!!! 3)如果上述两种方法仍然没解决,可以试一试笨办法: 出现此类问题,主要是 /usr/local/lib/目录下找不到libcudart.so.8.0这个文件, 该文件可以在usr/

caffe:mac10.12安装caffe的步骤

前言:caffe官网中安装caffe时涉及的东西比较多而杂乱,其实有些东西可以忽略掉. 步骤: mac10.12本身自带了python2.7,可以使用系统自带的python 1.安装Homebrew: 在终端中输入  ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)" 2.安装caffe需要用到的一些依赖库,在终端中依次输入下面几行 brew install

Caffe + Ubuntu 14.04 64bit + CUDA6.5 + 无GPU 配置

官网: http://caffe.berkeleyvision.org/installation.html#compilation 参考网站: http://www.cnblogs.com/dupuleng/articles/4213834.html http://www.cnblogs.com/empty16/p/4793404.html ------------------------------------------------------------------------------

Caffe + Ubuntu 14.04 + CUDA 6.5 新手安装配置指南

洋洋洒洒一大篇,就没截图了,这几天一直在折腾这个东西,实在没办法,不想用Linux但是,为了Caffe,只能如此了,安装这些东西,遇到很多问题,每个问题都要折磨很久,大概第一次就是这样的.想想,之后应用,应该还会遇到很多问题吧,不过没办法了,骑虎难下!!这里有个建议是,如果将来要做大数据集,最好事先给Linux留多点空间,比如Imagenet,估计500G都不为过.另外,请阅读完,至少一个部分再进行动手操作,避免多余的工作,写作能力有限,尽请见谅.        这篇安装指南,适合零基础,新手操

Caffe + Ubuntu 15.04 + CUDA 7.5 新手安装配置指南

Caffe + Ubuntu 15.04 + CUDA 7.5 新手安装配置指南 特: 0. Caffe 官网地址:http://caffe.berkeleyvision.org/ 1. 本文为作者亲自实验完成,但仅限用于学术交流使用,使用本指南造成的任何不良后果由使用者自行承担,与本文作者无关,谢谢!为保证及时更新,转载请标明出处,谢谢! 2. 本文旨在为新手提供一个参考,请高手勿要吐槽,有暴力倾向者,请绕道,谢谢! 3. 本文使用2015年11月8日下载的caffe-master版本,运行平

Caffe + Ubuntu 15.04 + CUDA 7.0 安装以及配置

作为小码农的我,昨天就在装这个东东了,主要参考第一篇博文,但是过程发现很多问题,经过反反复复,千锤百炼,终于柳暗花明,我把这个caffe给搞定了,是故,我发布出来,后之来者,欲将有感于斯文~ 本分分为四个部分,在Ubuntu上调试运行成功,第一部分:nVidia驱动和CUDA Toolkit的安装和调试:第二部分 Python安装和调试:第三部分 Matlab安装和调试:第四部分 Caffe的安装和测试. 第一部分:nVidia驱动和CUDA Toolkit的安装和调试 这里以CUDA 7.0为

Caffe+Ubuntu 14.04.4+CUDA7.5安装

ubuntu 14.04安装 先到官网下载ubuntu14.04,网址:http://www.ubuntu.com/download/desktop ubuntu-14.04.4-desktop-amd64.iso 参考:Ubuntu14.04 安装及使用:[1]制作安装U盘 制作安装U盘 然后参考:Ubantu14.04安装教程 安装系统 分区: boot 200M 主分区 / 50000M swap 4000M home 剩余M cuda7.5安装 cuda7.5下载:地址 https://