美文网首页
Caffe 安装与测试(包含CUDA)

Caffe 安装与测试(包含CUDA)

作者: NO_OcaNE | 来源:发表于2018-10-10 00:04 被阅读0次

    设备:

    Ubuntu 16.04

    python 2.7(anaconda) python —version

    cuda 9.0 nvcc -V

    cuDNN 7101 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

    openCV 2.4.9.1 pkg-config --modversion opencv

    caffe主页:

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

    主要步骤:

    安装依赖包

    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

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

    参考:  https://blog.csdn.net/u012235003/article/details/54575758/

    2.1 检查GPU的信息

    lspci | grep -i nvidia

    2.2 检查是否安装gcc

    gcc - - version

    2.3 检查kernel版本

    sudo apt-get install linux-headers-$(uname -r)

    2.4 禁用nouveau 第三方显卡驱动

    2.4.1  检查nouveau运行状态

    lsmod | grep nouveau

    2.4.2 在/etc/modprobe.d 中创建黑名单blacklist-nouveau.conf

    sudo  touch  blacklist-nouveau.conf

    sudo  chmod a+w+r blacklist-nouveau.conf  (给文件可读可写的权限)

    vim blacklist-nouveau.conf

    文件内容是:

    blacklist nouveau

    options nouveau modeset=0

    2.4.3 运行命令

    sudo update-initramfs -u

    2.4.4 重启设备

    2.4.5 检查nouveau运行状态

    2.5 下载CUDA

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

    选择runfile下载[注:之后可以按照官网安装方式安装CUDA]

    2.6 安装依赖包

    sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

    2.7 开始安装

    2.7.1.重启电脑 在进入到登录界面时候,按住Ctrl+Alt+F1,进入到text mode,登录账号 2.7.2.关闭图形界面 终端命令:

    sudo service lightdm stop

    2.7.3.切换到cuda文件目录 cd到下载好的cuda目录,例如我的cuda包在/home/tony/cuda中 再ls查看cuda名字 2.7.4.给cuda可执行的权限

    sudo chmod a+x cuda_7.5.18_linux.run

    2.7.5.安装步骤

    sudo sh cuda_7.5.18_linux.run

    (a)会先有个阅读声明,一直按D即可,然后accept (b)第一个选项install nvidia accelerated Graphics Driver (y) (c)第二个选项install the OpenGL libraries 双显卡(指的是集成显卡+独立显卡)选n,单卡(如果没有集成显卡,只有一个或多个支持GPU的显卡)选y。双显卡选y的话,会出现黑屏或登录界面无限循环的问题。 (d)后面的选项都是yes,或者按默认路径即直接按回车即可。 

    2.7.6.若出现下列显示 上面无警告和报错

    Driver :Installed

    Toolkit :Installed in /usr/local/cuda-7.5

    Samples :Installed in /home/tony

    即安装暂时成功。

    2.7.7.输入

    sudo service lightdm start

    重新启动图形化界面 Alt + ctrl +F7,返回到图形化登录界面,输入密码登录。 如果能够成功登录,则表示不会遇到循环登录的问题,基本说明cuda的安装成功了。

    2.7.8.搭配环境 终端输入

    echo 'export PATH=/usr/local/cuda-7.5/bin:$PATH' >> ~/.bashrc

    echo 'export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

    source ~/.bashrc

    重启ubuntu

    2.8 测试cuda安装结果

    2.8.1.检查路径 ~/dev 下 有无存在名为 nvidia* (以nvidia开头)的多个文件(device files) 若无,安装错误,见解决篇。 

    2.8.2.检查 CUDA Toolkit是否安装成功 终端输入 :

    nvcc -V

    会输出CUDA的版本信息(V要大写) 

    2.8.3.编译samples例子 进入到Samples安装目录,然后在该目录下终端输入make,等待十来分钟。 

    2.8.4.编译完成后测试 可以在Samples里面找到bin/x86_64/linux/release/目录,并切换到该目录 运行deviceQuery程序,sudo ./deviceQuery 查看输出结果,重点关注最后一行,Pass表示通过测试

    3 安装BLAS

    conda install blas

    4 安装些奇怪的东西(gflag, glog, lmdb)

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

    # glog

    wget https://github.com/google/glog/archive/v0.3.3.tar.gz

    tar zxvf v0.3.3.tar.gz

    cd glog-0.3.3

    ./configure

    make && make install

    # gflags

    wget https://github.com/schuhschuh/gflags/archive/master.zip

    unzip master.zip

    cd gflags-master

    mkdir build && cd build

    export CXXFLAGS="-fPIC" && cmake .. && make VERBOSE=1

    make && make install

    # lmdb

    git clone https://github.com/LMDB/lmdb

    cd lmdb/libraries/liblmdb

    make && make install

    5 下载Caffe

    https://github.com/BVLC/caffe

    6 编译Caffe

    cp Makefile.config.example Makefile.config

    # Adjust Makefile.config (for example, if using Anaconda Python, or if cuDNN is desired)

    make all

    make test

    make runtest

    7 遇到的典型问题汇总

    7.1 Compile Caffe 时vcc fatal : Unsupported gpu architecture 'compute_20'

    解决办法:

    将Makefile.config中不支持的CUDA_ARCH注释掉

    7.2 cannot find hdf5

    解决办法:

    在Makefile.config中

    INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/includeLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib

    修改为:

    INCLUDE_DIRS :=  $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

    //这是因为ubuntu16.04的文件包含位置发生了变化,尤其是需要用到的hdf5的位置,所以需要更改这一路径然后:

    cd /usr/lib/x86_64-linux-gnu

    软链接:

    sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so

    sudo ln -s libhdf5_serial_hl.so.10.0.2 libhdf5_hl.so

    7.3 在make runtest 的时候遇到段错误 core dumped

    查看warning信息:

    Warning! ***HDF5 library version mismatched error***

    The HDF5 header files used to compile this application do not match the version used by the HDF5 library to which this application is linked.Data corruption or segmentation faults may occur if the application continues.

    This can happen when an application was compiled by one version of HDF5 but linked with a different version of static or shared HDF5 library.

    You should recompile the application or check your shared library related settings such as 'LD_LIBRARY_PATH'.

    You can, at your own risk, disable this warning by setting the environment variable 'HDF5_DISABLE_VERSION_CHECK' to a value of ‘1'. Setting it to 2 or higher will suppress the warning messages totally. Headers are 1.10.2, library is 1.8.16

    故障分析:

    由于安装了anaconda,在安装包的过程中,出现了两个版本的HDF5,发生冲突。

    解决办法:

    step 1.给anaconda赋权,(一开始由于未赋予足够权限,导致了删除、更新包过程出现错误),

    sudo chmod -R 777 anaconda2/

    step 2. 将anaconda中hdf5版本改变到系统对应版本

    conda install -c anaconda hdf5=1.8.16

    即更新下列包:

    h5py: 2.8.0-py27h989c5e5_3 --> 2.6.0-np111py27_1

    hdf5:  1.10.2-hba1933b_1    --> 1.8.16-0       

    numpy: 1.15.1-py27h1d66e8a_0 --> 1.11.3-py27hcd700cb_6

    pytables: 3.4.4-py27ha205bf6_0  --> 3.2.2-np111py27_4   

    scipy:    1.1.0-py27hfa4b5c9_1  --> 1.1.0-py27hd20e5f9_0

    【注】论坛https://www.cnblogs.com/platero/p/4077934.html提供的解决方案是错的


    Error-7.3 差不多花了我5个小时,中间解决思路错了,多费了很多无用功,好在最后Mission Complete!

    相关文章

      网友评论

          本文标题:Caffe 安装与测试(包含CUDA)

          本文链接:https://www.haomeiwen.com/subject/vseuaftx.html