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