centos7.4+caffe+CPU(only) 配置(留给自己看!)

本文记录了在CentOS7.4上配置仅支持CPU的Caffe的详细过程,包括VMware虚拟机的设置、环境配置、OpenBLAS的安装以及Caffe的编译步骤,旨在解决依赖问题和加速训练。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考: https://blog.youkuaiyun.com/iotlpf/article/details/74669503

            https://blog.youkuaiyun.com/qq1326702940/article/details/82322079

            https://blog.youkuaiyun.com/xiaojun111111/article/details/59042673

2019/8/29 工作需要在centos7.4上配置caffe(CPU only),踩了一天的坑,写个博客给自己看。

一、准备工作:

1)下载 VMware 12.0.1 (专业版永久许可证密钥:5A02H-AU243-TZJ49-GTC7K-3C61N)

https://download3.vmware.com/software/wkst/file/VMware-workstation-full-12.1.0-3272444.exe

2)在VMware12中首先配置centos7.4虚拟机

下载资源:

直接使用VMware12打开即可

二、配置环境:

1)centos7.4中常用yum指令进行下载及安装,它是CentOS中的Shell前端软件包管理器,基于RPM包管理,能够从指定的服务器自动下载RPM包并且安装,可以自动处理依赖性关系,并且一次安装所有依赖的软件包,无须繁琐地一次次下载、安装。

sudo yum install epel-release 
sudo yum update 

2) 按需安装其他软件(GCC&G++推荐安装)

sudo yum install gcc gcc-c++
sudo yum install git vim python-devel python-pip

3) 根据git上caffe for centos安装手册,安装依赖库

sudo yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel
sudo yum install gflags-devel glog-devel lmdb-devel
sudo yum install atlas-devel
sudo yum install python-devel

4) 安装OpenBLAS

使用centos自带的atlas并按照caffe官网的安装时会报错(cannot find -lcblas & -latlas)

*这种方法屡试不爽,我在docker容器(Cetnos7)中配置caffe用这种方法也成功了

除此之外,OpenBLAS可以在caffe训练过程中加速

git clone https://github.com/xianyi/OpenBLAS.git
cd OpenBLAS
make -j4
make install

三、编译caffe

1) 下载caffe

git clone https://github.com/BVLC/caffe.git
cd caffe
cp Makefile.config.example Makefile.config
vim Makefile.config

2)修改Makefile.config文件:

注意,按照自己的安装需求,首先将

# CPU_ONLY := 1

改为

CPU_ONLY := 1

然后,由于使用的是OpenBLAS,将

BLAS := atlas

改为

BLAS := open

最后需要配置BLAS路径(使用make install默认路径)

找到

#BLAS_INCLUDE :=/path/to/your/blas

#BLAS_LIB :=/path/to/your/blas

改为

BLAS_INCLUDE := /opt/OpenBLAS/include

BLAS_LIB := /opt/OpenBLAS/lib

*这里opt/OpenBLAS是默认的OpenBLAS的安装目录

3) 编译caffe

cd caffe
make clean
make all -j8
make test
make runtest

编译成功

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值