Ubuntu18.04+Anaconda3快速搭建Caffe环境

时间:2024-05-19 16:18:11

        如何在Ubuntu安装Anaconda和配置环境变量这里不详细叙述,可以参考其他文章。

        网上大部分都是都是在Ubuntu14、16下安装Caffe的,安装过程复杂,问题出现的也比较多,所以这里选择了Ubuntu18(17也可以),在Ubuntu18.04中,只需要一行代码就可以安装好Caffe。原因就是在Ubuntu 17.04以上的系统全都包含了所有的依赖而且直接提供了编译好的Caffe,可以自行到Ubuntu官网查看。打开终端输入

sudo apt install caffe-cpu

就可以了,因为我没有NVIDIA的显卡,所以在这里安装的是cpu版,如果想安装gpu的输入命令sudo apt install caffe-cuda即可,不需要那些dependencies,也不需要配置Makefile.config,安装gpu版的还需要安装CUDA和CUDNN,网上也有很多教程可以参考,这里不过多叙述。

        Ubuntu默认已经安装了Python2和Python3,在没有安装Anaconda的时候,默认使用的Python3,在终端输入Python或者Python3都可以查看Python的版本为同一个。这时候安装的caffe目录就位于/usr/lib/python3/dist-packages/caffe

Ubuntu18.04+Anaconda3快速搭建Caffe环境在没有安装Anaconda的情况下,在终端输入Python,进入python解释器后输入import caffe回车,如果出现下边的错误

ImportError: No module named google.protobuf.internal

解决方法就是安装protobuf模块,退出python解释器,终端输入下边的命令,等待安装完毕即可。

sudo pip install protobuf

        如果安装了Anaconda3,并配置了Anaconda的环境变量,在终端输入python或者python3用的都是Anaconda的python解释器,进入解释器后倒入caffe模块,会出现下边的错误:

>>>import caffe
   Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
   ImportError: No module named caffe

这种情况一般是没有把caffe中的和python相关的内容的路径添加到python的编译路径中。可以通过在代码中添加caffe路径,,但是治标不治本,每次都需要添加才可以。也可以直接在配置文件中添加caffe的python解释器,不过也比较麻烦,这里提供一个简单的方法,可以通过将在/usr/lib/python3/dist-packages/caffe中的caffe文件夹复制进Anaconda的/anaconda3/lib/python3.6/site-packages目录下即可,因为Anaconda使用的模块都在此文件夹下边

Ubuntu18.04+Anaconda3快速搭建Caffe环境此时在终端进入python解释器,输入caffe就没有问题了,回车后没有提示错误问题就可以了(Anaconda也需要提前安装好protobuf模块,直接通过sudo pip install protobuf安装)。

Ubuntu18.04+Anaconda3快速搭建Caffe环境