Ubuntu14.04下编译pycaffe并绘制网络结构图

小灰灰 2022-06-06 13:17 140阅读 0赞

# 前言 #

由于课题需要，编译了Caffe的python接口。下面的博客内容就将整个过程和遇到的问题记录下来，若是各位看官遇到了相似的情况可能会给你带来帮助。

# 1. 环境准备 #

sudo apt-get update
    sudo apt-get install python-pip puthon-dev python numpy
    sudo apt-get install python-pip python-dev python-numpy
    sudo apt-get install gfortran
    sudo pip install -r $caffe_root/python/requirements.txt

说明：$caffe\_root是指caffe的目录，也就是解压的目录。在进行上面安装，当到安装IPython（对Python2.7.x）的时候可能会出现限免的情况导致安装不成功。

IPython 6.0+ does not support Python 2.6, 2.7, 3.0, 3.1, or 3.2.
        When using Python 2.7, please install IPython 5.x LTS Long Term Support version.
        Beginning with IPython 6.0, Python 3.3 and above is required.
        
        See IPython `README.rst` file for more information:
        
            https://github.com/ipython/ipython/blob/master/README.rst
        
        Python sys.version_info(major=2, minor=7, micro=6, releaselevel='final', serial=0) detected.
        Your pip version is out of date, please install pip >= 9.0.1. pip 1.5.4 detected.
        
        Complete output from command python setup.py egg_info:    
    
    IPython 6.0+ does not support Python 2.6, 2.7, 3.0, 3.1, or 3.2.
    When using Python 2.7, please install IPython 5.x LTS Long Term Support version.
    Beginning with IPython 6.0, Python 3.3 and above is required.

这里是说我们安装的IPython的版本太高了，需要安装IPython5左右的。这里就有两种办法去解决：  
**办法1：源码安装**

wget https://codeload.github.com/ipython/ipython/zip/5.x -o ipython-5.x.zip
    unzip ipython-5.x.zip
    cd ipython-5.x
    # The [test] extra ensures test dependencies are installed too:
    pip install .[test]
    #Do not invoke setup.py directly as this can have undesirable consequences for further upgrades. We do not recommend using easy_install either.
    ln -sf /usr/local/bin/ipython /usr/bin/ipython
    ipython -V

**方法二：不知道那个版本可以用，那就让系统来做**

sudo pip install ipython==8888 #现指定一个不存在的版本以查看可用版本，发现5版本中有5.3.0（你的可能不一样）

就会出现这些可用的版本供你选择  
![选择版本][SouthEast]  
之后我们选择一个可用的版本就好了。ref：[http://www.jianshu.com/p/8779eac7f313][http_www.jianshu.com_p_8779eac7f313]

sudo pip install ipython==5.0.0
    sudo pip install pydot

# 2. 编译pycaffe #

这里的编译就比较简单了

make clean
    make -j4
    make pycaffe

# 3. 绘制网络结构图 #

这里以绘制mnist使用的网络结构为例进行绘制网络结构。当前命令所在的目录在$caffe\_root/python/下

python draw_net.py ../examples/mnist/lenet.prototxt mnist.png

可能会出现如下错误

Drawing net to mnist.png
    Traceback (most recent call last):
      File "draw_net.py", line 58, in <module>
        main()
      File "draw_net.py", line 54, in main
        phase)
      File "/home/sucker/Desktop/install_pakages/caffe-1.0/python/caffe/draw.py", line 244, in draw_net_to_file
        fid.write(draw_net(caffe_net, rankdir, ext, phase))
      File "/home/sucker/Desktop/install_pakages/caffe-1.0/python/caffe/draw.py", line 223, in draw_net
        return get_pydot_graph(caffe_net, rankdir, phase=phase).create(format=ext)
      File "/usr/local/lib/python2.7/dist-packages/pydot.py", line 1883, in create
        prog=prog))
    Exception: "dot" not found in path.

**解决办法**

sudo apt-get install graphviz

生成的网络结构图  
![网络结构][SouthEast 1]

[SouthEast]: /images/20220606/ff8453f38eef4299b7a3ef9cb6ea64fa.png
[http_www.jianshu.com_p_8779eac7f313]: http://www.jianshu.com/p/8779eac7f313
[SouthEast 1]: /images/20220606/8f7a6ec493494614aa3b2fc438dde950.png