首页 >> 资料手册 >> 计算软件
Caffe使用说明
2017-02-28

Caffe是一个清晰而高效的深度学习框架。目前已经安装到ChinaGrid上面。用户可以直接使用预安装好的Caffe,也可以在自己的目录里面自行编译使用。

  • 预安装的Caffe位于/opt/caffe,使用前需要设置的相关环境变量如下:
    #CUDA 8.0
    export PATH=/opt/cuda-8.0/bin:$PATH
    export LD_LIBRARY_PATH=/opt/cuda-8.0/lib64:$LD_LIBRARY_PATH
    export INCLUDE=/opt/cuda-8.0/include:$INCLUDE
    export MANPATH=/opt/cuda-8.0/doc/man:$MANPATH
    #boost 1.63.0
    export LD_LIBRARY_PATH=/opt/boost/1.63.0/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/boost/1.63.0/include:$INCLUDE
    #glog for caffe
    export LD_LIBRARY_PATH=/opt/glog/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/glog/include:$INCLUDE
    #gflags for caffe
    export LD_LIBRARY_PATH=/opt/gflags/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/gflags/include:$INCLUDE
    #lmdb for caffe
    export PATH=/opt/lmdb/bin:$PATH
    export LD_LIBRARY_PATH=/opt/lmdb/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/lmdb/include:$INCLUDE
    export MANPATH=/opt/lmdb/share/man:$MANPATH #caffe
    export LD_LIBRARY_PATH=/opt/caffe/lib:$LD_LIBRARY_PATH
    export PATH=/opt/caffe/tools:$PATH

  • Caffe源代码位于/opt/src/caffe/caffe.chinagrid.tgz,编译前请根据需要修改Makefile.config等相关文件,并参考以下环境变量设置:
    #CUDA 8.0
    export PATH=/opt/cuda-8.0/bin:$PATH
    export LD_LIBRARY_PATH=/opt/cuda-8.0/lib64:$LD_LIBRARY_PATH
    export INCLUDE=/opt/cuda-8.0/include:$INCLUDE
    export MANPATH=/opt/cuda-8.0/doc/man:$MANPATH
    #boost 1.63.0
    export LD_LIBRARY_PATH=/opt/boost/1.63.0/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/boost/1.63.0/include:$INCLUDE
    #glog for caffe
    export LD_LIBRARY_PATH=/opt/glog/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/glog/include:$INCLUDE
    #gflags for caffe
    export LD_LIBRARY_PATH=/opt/gflags/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/gflags/include:$INCLUDE
    #lmdb for caffe
    export PATH=/opt/lmdb/bin:$PATH
    export LD_LIBRARY_PATH=/opt/lmdb/lib:$LD_LIBRARY_PATH
    export INCLUDE=/opt/lmdb/include:$INCLUDE
    export MANPATH=/opt/lmdb/share/man:$MANPATH

    如果使用过程中出现问题,请和超算中心联系。Email: sccadmin@ustc.edu.cn ,QQ群:8355136,电话:63602248。

    参考资料
    1. Caffe主页: http://caffe.berkeleyvision.org

  • Copyright 2009 中国科学技术大学超级计算中心 All Rights Reserved