在Ubuntu上安装caffe

Caffe是一款优秀的深度学习编程框架,这段时间因为需要使用LSTM、CNN等,因此尝试在有GPU的主机上安装Caffe,系统是ubuntu。

首先从官网下载Caffe的最新版源码包,在安装之前,先需要更新部分系统依赖:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

在安装Caffe之前,需要安装cuda和Nvidia驱动,由于主机上的GPU有点问题,因此先略过,在makefile.config里面,把CPU_ONLY=1的注释先取消掉,就可以用CPU的模式来处理

然后需要安装BLAS库,这里我们使用的是OpenBlas,而Caffe默认的是Atlas,因此需要修改配置文件,首先cp Makefile.config.example Makefile.config,然后修改这个默认的配置文件:vim Makefile.config
默认选项是BLAS := atlas,我们修改成BLAS:= open即可,另外保险起见,可以把选项下面的路径也设置为OpenBlas的对应路径

然后就是输入make all安装,可以加入-j4之类的,多线程编译加快速度
然后运行make test,再运行make runtest,出现问题ibopenblas.so.0: cannot open shared object file
解决办法是修改profile文件,加入LD_LIBRARY_PATH=$ LD_LIBRARY_PATH:/home/hadoop/whx/OpenBLAS/lib/

然后make runtest,测试Caffe各部件的运行情况,如果最终全部OK,然后显示PASSED,说明安装完成。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值