zoukankan      html  css  js  c++  java
  • ubuntu14.04安装caffe+python接口

    1 ubuntu系统安装

    ubuntu 14.04下载地址:http://www.releases.ubuntu.com/14.04/

    可以直接点击以下地址直接下载

    http://www.releases.ubuntu.com/14.04/ubuntu-14.04.5-desktop-amd64.iso

    2 caffe环境安装

    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
    BLAS: sudo apt-get install libatlas-base-dev

    Python: sudo apt-get install python-dev python-pip

    ubuntu 14.04 :sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

    CUDA: 使用的cuda7.5版本

    sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
    sudo apt-get update
    sudo apt-get install -y cuda

    设置cuda环境:

    在/etc/profile文件最后添加CUDA环境变量:

    sudo gedit /etc/profile

    PATH=/usr/local/cuda/bin:$PATH

    export PATH   

    保存后, 执行下列命令, 使环境变量立即生效

    source /etc/profile

    同时需要添加lib库路径: 在 /etc/ld.so.conf.d/加入文件 cuda.conf

    sudo gedit /etc/ld.so.conf.d/cuda.conf

    内容如下:

    /usr/local/cuda/lib64 

    保存后,执行下列命令使之立刻生效

    sudo ldconfig 

    CUDNN:使用4.0版本

    tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz  
    cd cuda  
    sudo cp lib64/lib* /usr/local/cuda/lib64/  
    sudo cp include/cudnn.h /usr/local/cuda/include/   
    
    cd /usr/local/cuda/lib64/
    sudo chmod +r libcudnn.so.4.0.7
    sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4
    sudo ln -sf libcudnn.so.4 libcudnn.so
    sudo ldconfig

    CUDNN:使用5.0版本

    tar -zxvf cudnn-7.5-linux-x64-v5.0-ga.tgz  
    cd cuda  
    sudo cp lib64/lib* /usr/local/cuda/lib64/  
    sudo cp include/cudnn.h /usr/local/cuda/include/   
    
    cd /usr/local/cuda/lib64/
    sudo chmod +r libcudnn.so.5.0.5
    sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
    sudo ln -sf libcudnn.so.5 libcudnn.so
    sudo ldconfig

    3 python 接口编译

    安装git和下载caffe源码:

    sudo apt-get install build-essential cmake git pkg-config
    git clone https://github.com/BVLC/caffe.git

    下载python依赖环境 

    sudo su 
    sudo apt-get install gfortran
    pip install --upgrade pip
    for req in $(cat requirements.txt); do pip install $req; done 
    // 推荐使用国内镜像安装,你懂的,速度快
    for req in $(cat requirements.txt); do pip install $req -i http://pypi.douban.com/simple; done 
    for req in $(cat requirements.txt); do pip install $req -i https://pypi.tuna.tsinghua.edu.cn/simple; done 

    编译:先拷贝一份Makefile.config文件,取消一下注释

    cp Makefile.config.example  Makefile.config

    # cuDNN acceleration switch (uncomment to build with cuDNN).
    USE_CUDNN := 1

    # Uncomment to support layers written in Python (will link against Python libs)
    WITH_PYTHON_LAYER := 1

    使用如下命令编译caffe和python接口

    make -j
    make py

    编译 pycaffe时报错:fatal error: numpy/arrayobject.h没有那个文件或目录

    其实numpy已经是安装的,anaconda2里面有,Python中import numpy也没有问题,但就是在此处报错,解决方法:
    sudo apt-get install python-numpy

    4 使用spyder调用python接口和kdevelop调用c++

    安装

    sudo pip install spyder
    sudo apt-get install kdevelop
  • 相关阅读:
    Spring Boot入门
    Spring MVC文件上传和下载
    Spring MVC异常处理
    SpringMVC
    linux(2)
    linux(1)
    白盒测试
    LoadRunner(8)
    LoadRunner(7)
    LoadRunner(6)
  • 原文地址:https://www.cnblogs.com/linyuanzhou/p/6136912.html
Copyright © 2011-2022 走看看