首先安装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
网友评论