在网上的一些教程的基础上,加上自己调试了一个下午终于成功安装了,可能大部分与网上的相同,但是加了一些我在安装时遇到的小问题。
这几天因为要参加一个比赛,所以急着给电脑装caffe,不多说了,进入正题:
第一步:(首先这是针对不需要使用GPU加速的同学参考的文章,所以需要配置GPU的请参考其他文章)
首先更新一下apt-get:
$sudo apt-get update
第二步:安装一些必须的依赖包
$sudo apt-get install build-essentials
第三步:安装ATLAS
这里,网上说caffe的BLAS有很多种选择ATLAS,MKL,OpenBLAS。因为看到网上有的人说不用GPU加速的就要安装ATLAS,别问我这个ATLAS是干嘛的。。。我只知道这是某种依赖包,必须安装。
$sudo apt-get install libatlas-base-dev
第四步:安装opencv
这个比较简单,https://github.com/jayrambhia/Install-OpenCV这是github网上的源码,下面直接有教程。为了方便,我这里就说一下吧。进入网站后,下载压缩包,直接解压。然后在终端利用cd进入你自己对应的系统文件内,然后输入:
$sudo chmod +x *
之后输入:
$./opencv_latest.sh
第五步:安装一些其他的依赖项
$sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler
第六步:安装caffe
这是github上的源码网站:https://github.com/BVLC/caffe,下载压缩包然后解压,在终端中利用cd进入caffe-master目录下,然后输入命令:
$sudo cp Makefile.config.exmaple Makefile.config
利用这个命令将Makefile.config.example中的内容复制下来,接下来要对Makefile.config中的内容进行配置。
利用Ubuntu中的vim命令从终端进入文件进行改动:
$vim Makefile.config
如果出现问题,应该是没有安装vim,利用sudo apt-get install vim安装一下就行了。进入了Makefile.config中
这是我配置好的Makefile.config文件,里面的USE_CUDNN和CPU_ONLY这两行原来是被注释掉的,也就是前面是有#符号的,以为我们使用的只是CPU,所以将CPU_ONLY :=1这一行前面的井号去掉,等号右边的1意思也就是说只使用CPU(必须保证等于1)。
然后和其他教程不同,我在这个文档里面还改了一下其他的地方,因为直接按照网上的很多教程来做,最后会出错,所以我自己调试了一下里面其他的语句。
CUDNN好像是英伟达公司用来GPU加速的库,所以说白了就是和GPU有关的东西,所以这里我把他关掉了,也就时令他等于0,并且去掉了前面的#。
将文档往下翻,可以看到有一个是CUDA_DIR和CUDA_ARCH,因为我看到这两个和CUDA都有关,所以肯定都是关于GPU的设置,所以我就把这两句话给注释掉了,前面都加上了#(图中没有加#,因为方便大家找到那两句话。)。最后一个需要改的地方就是BLAS :=这句话了。在之前我们安装的时ATLAS,所以这里等号右边应该也是atlas。用vim将上面的都修改好之后,就可以保存退出了(想修改vim打开的文档输入i就可以了。修改的时候vim处于插入模式,按esc退出然后输入:wq就可以保存退出了)。
第七步:配置安装caffe
此时终端的路径应该还保持在上一步中,就是caffe-master下,输入:
$make all -j4
这里j后面的数字就是你的电脑核数,根据自己电脑的核数修改,也可以直接不用-j4这个参数。
然后再输入:
$make test
最后输入:
$make runtest
如果这个过程没有出错,那么caffe应该算是安装和配置好了,就差在python中直接调用了(如果出错的话,查看错误原因是不是因为少装了依赖的包,可以试着使用sudo apt-get install xxx命令安装一下,或者可以联系我一起讨论一下)。到现在为止,应该只是在电脑中安装了caffe,可以直接在命令行中使用caffe。但是还不能在python下直接import。所以还需要最后一步。
最后一步:添加caffe目录到python环境变量中
这时候终端中的路径还是应该在caffe-master下,输入:
$sudo make pycaffe
<pre name="code" class="html" style="font-size: 18px;">$export PYTHONPATH=$PYTHONPATH:你电脑上的caffe目录/python
这样就完全结束了安装。这时候进入python,输入import caffe,如果没有出错,就添加成功了。
http://www.ithao123.cn/content-1759601.html这是我参考的网络教程,但是在我的电脑上最后出错了,所以我在他的上面改了一些,大家也可以参考一下这篇文章。