您的位置:首页 > Web前端

深度学习:ubuntu16.04环境下 caffe的搭建

2018-07-20 15:13 369 查看

caffe安装教程

注意事项

综合比较了windos,centos与ubuntu三个系统,个人觉得在ubuntu系统下安装caffe较好,并且推荐16.04版本的ubuntu(gcc4.7.7以上,支持C++11,才能编译更加畅通无阻),如果是虚拟机安装的Ubuntu系统,建议硬盘设置大一些,避免其他不必要的麻烦。

1.检查系统是否支持GPU,再选择你安装的版本

 通常情况下,gpu比cup处理更快,但一些电脑不支持gpu

输入  

  1.  lspci | grep -i nvidia 

结果 1不输出结果,说明不支持GPU

 

      2.输出结果,则支持GPU

      所以本文选择安装CPU版本

2.安装依赖

----------------------------------------------------------------------------------------------------

依次输入 sudo apt-get install libprotobuf-dev

               sudo apt-get install libleveldb-dev

               sudo apt-get install libsnappy-dev

               sudo apt-get install libopencv-dev

               sudo apt-get install libhdf5-serial-dev

               sudo apt-get install protobuf-compiler

               sudo apt-get install libgflags-dev

               sudo apt-get install libgoogle-glog-dev

               sudo apt-get install liblmdb-dev

               sudo apt-get install libatlas-base-dev

               sudo apt-get install  libboost-all-dev 

------------------------------------------------------------------------------------------------------

3.下载caffe

可以直接用Git下载,先安装git  :    sudo apt-get install git

选择你想要存放的位置,再执行以下命令  :git clone git://github.com/BVLC/caffe.git

4.编译caffe

 1.进入caffe目录:  cd /caffe

 2.对Makefile.config.example文件备份:  cp Makefile.config.example Makefile.config

 3.编辑文本 : sudo gedit Makefile.config      

           1.去掉CPU_ONLY前面的#号

            2.配置引用文件路径

    

 

4.编译

  1.   sudo make all
  2.   sudo make test sudo make runtest
  3.   sudo make runtest

 

 

5设置python接口

caffe支持的python版本较低,如果用python3设置接口,可能会出现不能导入的情况!

我的ubuntu版本是16.04 自带python版本是2.7,所以直接开始:

    1.安装pip

     sudo apt-get install python-pip

    2.安装fortran编辑器(gfrotran),因为安装scipy库时需要它

    sudo apt-get install gfortran

    3.在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息

    cd ~/caffe/python for req in $(cat requirements.txt); do pip install $req; done

    4.安装结束后验证

    sudo pip install -r requirements.txt

    5.加入环境变量

    sudo gedit ~/.bashrc

   在文件最后加入

   export PYTHONPATH=~/caffe/python:$PYTHONPATH

    注意这里的路径是你系统caffe安装de的路径

  7.更新配置

  sudo ldconfig

  8.编译接口

  cd ~/caffe/

  make pycaffe

 9.验证 python接口

  

阅读更多
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐