caffe2 环境搭建 (windows 10 pro + docker)

背景

背景不多说，想体验一下 caffe2。

于是需要在 windows 下安装一个 caffe2。

参考官方文档，安装方式有源码安装 和 Docker Images 两种。

先尝试了安装 ubuntu 虚拟机，在虚拟机中通过 Build From Source 安装。然后发现 官方文档 和 github 上的源码对应不上，已经比较老，基本不适用了。

于是本着初次使用，不要花费过多时间在环境搭建上的原则，尝试 Docker Image 方法。

步骤

安装 docker

见 windows 下安装 docker 的 官方文档。

简单来讲，就是下载 稳定版本的安装包，然后一路 next。

目前看，windows 下的 docker，是新建了一个名字叫做 MobyLinuxVM 的 HyperVM 虚拟机，在 Hyper-V 管理器中可以看到这个虚拟机。

安装 caffe2 镜像

设置 docker 镜像地址

由于官方源下载速度太慢，设置了一个国内 docker 镜像地址。

docker 中国官方镜像

在 daemon.json 配置文件中加入

{  "registry-mirrors": ["https://registry.docker-cn.com"]}

阿里云官方 Docker 镜像加速器

这里需要用户自己注册，然后使用系统分配前缀，最终如下：https://xxxxxx.mirror.aliyuncs.com。

我这里使用了网友提供的前缀，就不在文中提供了。

当然也可以使用国内别的源，参考国内 docker 仓库镜像对比

下载镜像

开始 -> 运行 -> powershell

找到镜像的名称，在 docker hub 上查看。这里使用了一个 minimum 版本的镜像 tag。
需要注意的是，文档上的镜像 tag 不是最新的，要以上面 docker hub 上的为准。

docker pull caffe2ai/caffe2:c2v0.8.1.cpu.min.ubuntu14.04

使用国内镜像的方式


docker pull registry.docker-cn.com/caffe2ai/caffe2:c2v0.8.1.cpu.min.ubuntu14.04

测试

docker run -it caffe2ai/caffe2:c2v0.8.1.cpu.min.ubuntu14.04 /bin/bash
python -m caffe2.python.operator_test.relu_op_test

测试后，会提示缺少两个包，于是安装上就可以了

apt-get install graphviz"
pip install pydot

安装 jupyter

pip install jupyter

保存当前的环境

先退出容器

exit

再提交改变

docker ps -a

docker commit 35f83fab45d2 caffe2ai/caffe2:fw

使用新容器进入

docker run -it caffe2ai/caffe2:fw /bin/bash

以 jupyter 方式进入

docker run -it -p 8888:8888 caffe2ai/caffe2:fw sh -c "jupyter notebook --no-browser --ip 0.0.0.0 /caffe2/caffe2/python/tutorials --allow-root"

通过浏览器打开提示的链接，如

http://localhost:8888/?token=xxxxxx

附录

• docker images 压缩方法
• docker 避免出现的 10 种情况