一种Caffe在CentOS系统下的环境搭建方法技术方案

技术编号:21659580 阅读:22 留言:0更新日期:2019-07-20 05:51
本发明专利技术提供的一种Caffe在CentOS系统下的环境搭建方法,采用确定硬件基础,并以root身份登录系统;在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;对安装驱动和必要环境进行测试,如报错则进行处理,如通过则进行下一步;运行caffe,对运行情况和运行结果进行验证的方案,解决现有技术中环境搭建步骤复杂,搭建过程中经常报错的技术问题,实现过程清晰明确简单,有利于环境搭建过程中提高工作效率的技术效果。

A Caffe Environment Building Method in CentOS System

【技术实现步骤摘要】
一种Caffe在CentOS系统下的环境搭建方法
本专利技术涉及新计算机领域,具体而言,涉及一种Caffe在CentOS系统下的环境搭建方法。
技术介绍
Caffe,全称ConvolutionalArchitectureforFastFeatureEmbedding。是一种常用的深度学习框架,主要应用在视频、图像处理方面的应用上。随着社会的发展,对视频、图像的处理速度与质量要求越来越高。近年来,GPU卡的应用越来越广泛,它可以大幅提高服务器的运算能力。Caffe深度学习框架是开发的趋势。Caffe框架以“层”为单位对深度神经网络的结构进行了高度的抽象,通过一些精巧的设计显著优化了执行效率,并且在保持高效实现的基础上不失灵活性。无论在结构、性能上,还是代码质量上,Caffe都是一款出色的开源框架,现有环境搭建步骤复杂,搭建过程中经常报错,令人不知所措。
技术实现思路
本专利技术提供一种Caffe在CentOS系统下的环境搭建方法,采用确定硬件基础,并以root身份登录系统;在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;对安装驱动和必要环境进行测试,如报错则进行处理,如通过则进行下一步;运行caffe,对运行情况和运行结果进行验证的方案,解决现有技术中环境搭建步骤复杂,搭建过程中经常报错的技术问题。本专利技术为解决上述技术问题而提供的这种Caffe在CentOS系统下的环境搭建方法,包括以下步骤,S1.确定硬件基础,并以root身份登录系统;S2.在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;S3.对安装驱动和必要环境进行测试,如报错则进行处理,如通过则进行下一步;S4.运行caffe,对运行情况和运行结果进行验证。进一步,所述S1步骤中所述硬件基础为可运行Centos7.3完整系统的硬件,所述硬件基础中包括NVIDIATESLAP4GPU卡并连接互联网,所述完整系统中包含serverwithgui包并关闭防火墙。进一步,所述S2步骤中所述必要的驱动包括网卡驱动、NVIDIATESLAP4显卡驱动,所述必要工具包括cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe。进一步,所述S2步骤中安装必要的网卡驱动时所有工具包都放在/gw文件夹下。进一步,所述S2步骤中安装cuda还包括以下步骤,a.在命令行中运行命令代码,Createafileat/etc/modprobe.d/blacklist-nouveau.conf#vi/etc/modprobe.d/blacklist-nouveau.conf;blacklistnouveauoptionsnouveaumodeset=0b.在命令行中运行命令代码,#dracut--force;c.在命令行中运行命令代码,#init6;d.在命令行中运行命令代码,#init3;e.在命令行中运行命令代码,#vi/boot/grub2/grub.cfg添加”nomodeset3”到内核引导里;f.在命令行中运行命令代码,#shcuda_9.2.148_396.37_linux.run--silentg.添加环境变量,在命令行中运行命令代码,#vi/etc/profileexportPATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}exportLD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}#source/etc/profile;h.查看GPU信息,在命令行中运行命令代码,#nvidia-smi。进一步,所述S2步骤中安装cmake为安装最新版本的cmake,安装cmake同时开启多线程编译。进一步,所述S2步骤中安装python同时安装依赖包。进一步,所述S2步骤中安装caffe还包括修改caffe的编译配置步骤,S21.去掉USE_CUDNN项的#符号;S22.去掉USE_OPENCV项、USE_LEVELDB项、USE_LMDB项的#符号,将USE_OPENCV项、USE_LEVELDB项、USE_LMDB项的值从0改为1;S23.去掉USE_CUDNN项的#符号;S24.去掉CUDA_ARCH项的前两行;S25.将BLAS项的值改为mkl;S26.修改python路径为对应路径;S27.去掉anacondapython路径前的#,并修改对应路径;S28.将PYTHON_LIB项的值修改为/usr/lib64;S29.去掉WITH_PYTHON_LAYER项的#符号;S210.修改INCLUDE_DIRS项和LIBRARY_DIRS项的相应值;S211.在命令行中运行命令代码,l.#vi/etc/profileexportPYTHONPATH=/gw/caffe-master/python:\$PYTHONPATH#source/etc/profile;S212.在命令行中运行命令代码,#makeall–j8进行编译,如果出现报错,则每次报错修改后,执行makeclean命令,再执行命令makeall–j8,编译前确保/etc/ld.so.conf、/etc/profile已添加相关内容ld.so.conf。进一步,所述S3步骤中所述进行处理为运行命令代码#condainstallpydot。进一步,所述S4步骤中所述对运行情况和运行结果进行验证包括验证是否产生四个文件和验证CPU模式下运行的时间是否比CPU模式下运行的时间短,所述文件包括mnist_train_lmdb,mnist_test_lmdb。本专利技术所具有的有益效果:本专利技术提供的这种方法,过程清晰明确,有利于环境搭建过程中提高工作效率。附图说明图1是本专利技术所述Caffe在CentOS系统下的环境搭建方法流程图。具体实施方式结合上述附图说明本专利技术的具体实施例。如图1所示,本专利技术为解决上述技术问题而提供一种Caffe在CentOS系统下的环境搭建方法,包括以下步骤,S1.确定硬件基础,并以root身份登录系统;S2.在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;S3.对安装驱动和必要环境进行测试,如报错则进行处理重复S3步骤本文档来自技高网...

【技术保护点】
1.一种Caffe在CentOS系统下的环境搭建方法,其特征在于:包括以下步骤,S1.确定硬件基础,并以root身份登录系统;S2.在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;S3.对安装驱动和必要环境进行测试,如报错则进行处理,如通过则进行下一步;S4.运行caffe,对运行情况和运行结果进行验证。

【技术特征摘要】
1.一种Caffe在CentOS系统下的环境搭建方法,其特征在于:包括以下步骤,S1.确定硬件基础,并以root身份登录系统;S2.在系统环境下,安装必要的驱动和必要工具,所述必要工具的安装顺序依次为cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe;S3.对安装驱动和必要环境进行测试,如报错则进行处理,如通过则进行下一步;S4.运行caffe,对运行情况和运行结果进行验证。2.根据权利要求1所述的Caffe在CentOS系统下的环境搭建方法,其特征在于:所述S1步骤中所述硬件基础为可运行Centos7.3完整系统的硬件,所述硬件基础中包括NVIDIATESLAP4GPU卡并连接互联网,所述完整系统中包含serverwithgui包并关闭防火墙。3.根据权利要求1所述的Caffe在CentOS系统下的环境搭建方法,其特征在于:所述S2步骤中所述必要的驱动包括网卡驱动、NVIDIATESLAP4显卡驱动,所述必要工具包括cuda、cuDNN、BLAS、cmake、protoBuffer、glog、gflags、hdf5、OpenCV、lmdb、leveldb、python、pip、Blas、Cblas、boost、caffe。4.根据权利要求3所述的Caffe在CentOS系统下的环境搭建方法,其特征在于:所述S2步骤中安装必要的网卡驱动时所有工具包都放在/gw文件夹下。5.根据权利要求1所述的Caffe在CentOS系统下的环境搭建方法,其特征在于:所述S2步骤中安装cuda还包括以下步骤,a.在命令行中运行命令代码,Createafileat/etc/modprobe.d/blacklist-nouveau.conf#vi/etc/modprobe.d/blacklist-nouveau.conf;blacklistnouveauoptionsnouveaumodeset=0b.在命令行中运行命令代码,#dracut--force;c.在命令行中运行命令代码,#init6;d.在命令行中运行命令代码,#init3;e.在命令行中运行命令代码,#vi/boot/grub2/grub.cfg添加”nomodeset3”到内核引导里;f.在命令行中运行命令代码,#shcuda_9.2.148_396.37_linux.run--silentg.添加环境变量,在命令行中运行命令代码,#vi/etc/profileexportPATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}...

【专利技术属性】
技术研发人员:刘亚萍
申请(专利权)人:深圳市国鑫恒宇科技有限公司
类型:发明
国别省市:广东,44

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1