zoukankan      html  css  js  c++  java
  • Ubuntu16.04 caffe安装记录

     Ubuntu16.04 caffe安装记录

    1.安装显卡驱动

    首先更新输入:

    sudo apt-get update
    sudo apt-get upgrade
    

    然后打开System Settings中Software&Updates

    如下选择,并点击Apply Changes。

     2.禁用nouveau

    编辑文件

    sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
    

    在打开的文件中写入:

    blacklist nouveau option nouveau modeset=0
    

    执行

    sudo update-initramfs -u
    

    重启,输入

    reboot
    

     3.安装依赖包

    依次输入:

    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
    
    sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
    
    sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
    
    sudo apt-get install git cmake build-essential
    

    4.配置环境变量

    打开配置文件:

    sudo gedit ~/.bashrc
    

    添加:

    export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
    
    export LD_LIBRARY_PATH=/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH 
    

    保存退出

     5.安装CUDA 8.0

    执行可执行文件:

    sudo sh cuda_8.0.61_375.26_linux.run --no-opengl-libs
    

     注意:安装过程中不要选择安装显卡驱动,其他选择yes即可。

    重启:

    reboot
    

    打开配置文件:

    sudo gedit ~/.bashrc
    

    添加环境变量:

    export PATH=/usr/local/cuda-8.0/bin:$PATH
    
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
    

    使配置生效:

    source ~/.bashrc
    

    6.验证CUDA 8.0

    cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
    
    sudo make
    
    ./deviceQuery
    

     7.安装cudnn

    下载cudnn,并且拷贝到/usr/local路径下:

    sudo cp /home/peiyuyang/cudnn-8.0-linux-x64-v6.0.tgz /usr/local 

    进入路径下:

    cd /usr/local
    

     

    解压,执行:

    sudo tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
    

    解压后,执行:

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
    sudo chmod a+r /usr/local/cuda/include/cudnn.h
    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
    

    8.脚本安装OpenCV-3.3.0

    解压opencv压缩包,进入opencv目录下:

    unzip opencv-3.3.0.zip
    cd opencv-3.3.0

    编译opencv:

    mkdir build
    
    cd build
    
    sudo cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
    
    sudo make -j8

     编译完成之后,安装:

    sudo make install

    验证安装成功:

    pkg-config --modversion opencv  
    

      

    9.安装caffe

    安装路径下:

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

    將user目录下所有文件权限改为当前用户的权限:

    chown -R uer.uer /home/user
    

    进入caffe路径,拷贝Makefile.config.example:

    cp Makefile.config.example Makefile.config
    

    修改Makefile.config文件:

    gedit Makefile.config
    

    修改:

    将
    #USE_CUDNN := 1
    修改成: 
    USE_CUDNN := 1
    
    将
    #OPENCV_VERSION := 3 
    修改为: 
    OPENCV_VERSION := 3
    
    将
    #WITH_PYTHON_LAYER := 1 
    修改为 
    WITH_PYTHON_LAYER := 1
    
    将 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文件:

    gedit Makefile
    

    修改:

    将
    NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
    修改为:
    NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
    
    将
    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
    

    修改host_config.h文件:

    sudo gedit /usr/local/cuda/include/host_config.h 
    

    修改:

    将
    #error-- unsupported GNU version! gcc versions later than 5.0 are not supported!
    改为
    //#error-- unsupported GNU version! gcc versions later than 5.0 are not supported!
    

    开始编译caffe:

    make all -j8
    

    编译成功后运行测试:

    make runtest -j8
    

    10.编译pycaffe

    安装依赖 :

    sudo apt-get install -y python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags cython ipython
    

    编译python接口:

    make pycaffe
    

    进入caffe/python路径下,或者将python路径添加到环境变量,输入:

    python
    import caffe

    caffe安装完成 。

     11.安装Anaconda

    安装解压后,进入解压目录,运行:

    bash Anaconda2-4.4.0-linux-x86_64.sh
    

    选择添加环境变量。

    使环境变量生效:

    source ~/.bashrc
    

    之后进行:

    sudo pip install easydict
    pip install opencv-python collecting opencv-python
    pip install protobuf sudo apt-get install python-protobuf conda install libgcc

     

  • 相关阅读:
    SpringBoot: 2.SpringBoot整合servlet(转)
    SpringBoot: 1.创建第一个SpringBoot项目(转)
    SqlServer:SqlServer(服务器磁盘监控,创建管理员账号分配权,添加链接服务器,查询CPU,查询内存)
    Log parser工具使用
    目录爆破工具
    安装腾讯QQ问题记录
    Sql Server 2017 安装问题记录
    reGeorg+Proxifier使用
    Swaks
    python2.7 安装pycrypto库报错
  • 原文地址:https://www.cnblogs.com/peiyuYang/p/7784787.html
Copyright © 2011-2022 走看看