美文网首页
服务器(多GPU)caffe安装和编译

服务器(多GPU)caffe安装和编译

作者: angerskon | 来源:发表于2017-05-18 22:01 被阅读420次

    refer to

    一. 前提

    多GPU交互在神经网络是常见的,所以在安装caffe之前需要安装NCCL,来保证多GPU之间的相互交流。
    多GPU,这里指的是2个及2个以上英伟达显卡,而不是笔记本中的集显和独显。

    二.安装NCCL

    **1.下载编译 **
    shell终端

    cd nccl
    make CUDA_HOME=/user/local/cuda-7.5   test #注意自己的cuda路径
    

    2.测试和配置环境变量
    shell终端

    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./build/lib
    ./build/test/single/all_reduce_test
    ./build/test/single/all_reduce_test 10000000
    make install
    

    注:make install 是自己添加,而官方原文没有。之所以这么加是因为在caffe 执行 cmake时候,cmake无法找到
    非deb安装软件的路径,所以添加make install 是为了能让cmake识别到路径。

    三.安装caffe

    安装依赖&下载caffe (略)
    编译

    cp Makefile.config.example Makefile.config
    gedit Makefile.config
    

    打开文本后,作出如下修改
    取消下面这些话的前面注释符号#

    USE_CUDNN := 1 
    USE_NCCL := 1
    ANACONDA_HOME := $(HOME)/anaconda   #这里我们使用Anaconda环境下的python
    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
    

    在下面这些语句中加上#注释符号

    #PYTHON_INCLUDE := /usr/include/python2.7 \
            # /usr/lib/python2.7/dist-packages/numpy/core/include
    #PYTHON_LIB := /usr/lib
    

    保存,退出,编译caffe

    sudo pip install -r caffe/python/requirements.txt
    cd caffe
    mkdir build
    cd build
    make all -j
    make install -j
    make runtest -j
    

    相关文章

      网友评论

          本文标题:服务器(多GPU)caffe安装和编译

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