美文网首页
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