如何在ubuntu下搭建深度学习caffe环境

        在ubuntu下搭建caffe环境是比较困难的,会遇到各种坑,当时不用害怕,遇到问题解决问题就行了。

1、安装git,命令如下

sudo apt-get install git

2、安装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
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

3、下载caffe源码

       首先在计算中计划好caffe的安装位置,我的计算机中安装的目录为

/home/bigmarshal/Documents/deep_learning/

进入到这个目录,在终端上执行下面的命令

git clone https://github.com/bvlc/caffe.git

成功执行完这一步后,源码就已经下载到计算机中了,在deep_learning文件夹下生成了一个caffe文件夹,就是caffe的源码。

4、配置编译参数

        这一步很简单,caffe的源码中已经带了一个配置文件,我们进入到caffe文件夹,执行下面的命令

cp Makefile.config.example Makefile.config

这个名字的作用是将示例的配置文件复制为使用的配置文件名字。

用gedit打开Makefile.config文件,在这个文件中,我们基本不用修改,在文件的上边有一行,如下所示

# CPU_ONLY := 1

这句的配置是配置是否只使用CPU模式,如果只使用CPU模式,则把前面的#号去掉,即解除这句代码的注释,初学者建议从CPU模式学起,因为GPU模式的环境搭建过程会有更多的坑。

        修改完之后,保存退出。

5、编译caffe

编译很简单,执行下面的命令。

make -j8

至此,我们完成了caffe的安装,看起来还比较简单,但实际上,按照这个步骤安装一次成功的概率很小,总会遇到各种坑,我的博客里记录了我在安装caffe时,遇到的各种问题及解决办法,可以作为参考。

发布了73 篇原创文章 · 获赞 5 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/bhniunan/article/details/104352598