美文网首页
ubuntu 虚拟机利用docker安装caffe cpu版

ubuntu 虚拟机利用docker安装caffe cpu版

作者: ccccxxxxcccc | 来源:发表于2019-03-11 17:37 被阅读0次

    首先安装docker

    sudo apt-get install docker.io

    安装完成之后pull一个caffe cpu版的镜像

    sudo docker pull elezar/caffe:cpu 

    之后验证是否成功

    sudo docker run -ti elezar/caffe:cpu caffe --version

    出现此页面代表成功

    也可以查看镜像

    sudo docker images

    之后进入docker

    sudo docker run -t -i elezar/caffe:cpu /bin/bash

    caffe在/opt/caffe/目录下

    使用exit可退出docker

    需要再次进入时可先使用sudo docker ps -a查看当前运行的容器,并复制容器号CONTAINER ID

    使用sudo docker start CONTAINER ID开启此容器

    使用sudo docker attach CONTAINER ID进入此容器

    cd /opt/caffe/目录下输入python,然后import caffe

    出现上述

    libdc1394 error: Failed to initialize libdc1394

    可使用如下方法解决

    ln -s /dev/null /dev/raw1394

    即可成功import caffe

    如果在import caffe时出现以下错误

    >>> import caffe

    Traceback (most recent call last):

      File "<stdin>", line 1, in <module>

      File "/opt/caffe/python/caffe/__init__.py", line 1, in <module>

        from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer

      File "/opt/caffe/python/caffe/pycaffe.py", line 15, in <module>

        import caffe.io

      File "/opt/caffe/python/caffe/io.py", line 2, in <module>

        import skimage.io

      File "/usr/local/lib/python2.7/dist-packages/skimage/__init__.py", line 158, in <module>

        from .util.dtype import *

      File "/usr/local/lib/python2.7/dist-packages/skimage/util/__init__.py", line 7, in <module>

        from .arraycrop import crop

      File "/usr/local/lib/python2.7/dist-packages/skimage/util/arraycrop.py", line 8, in <module>

        from numpy.lib.arraypad import _validate_lengths

    ImportError: cannot import name _validate_lengths

    可能是由于numpy版本过高引起的

    首先查看numpy版本

    重新安装较低版本numpy

    pip install numpy==1.15.0

    之后便可成功import caffe

    相关文章

      网友评论

          本文标题:ubuntu 虚拟机利用docker安装caffe cpu版

          本文链接:https://www.haomeiwen.com/subject/mrgbpqtx.html