caffe2环境搭建(windows 7 + docker)

版权声明:我是南七小僧,微信: to_my_love ,寻找人工智能相关工作,欢迎交流思想碰撞。 https://blog.csdn.net/qq_25439417/article/details/84285126

背景

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

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

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

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

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

步骤

安装码头

见windows下安装docker的  官方文档

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

目前看,窗口下的搬运工,是新建了一个名字叫做MobyLinuxVM的HyperVM虚拟机,在Hyper-V的管理器中可以看到这个虚拟机。

安装caffe2镜像

设置泊坞窗镜像地址

由于官方源下载速度太慢,设置了一个国内泊坞窗镜像地址。

搬运工中国官方镜像

在daemon.json配置文件中加入

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

阿里云官方Docker镜像加速器

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

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

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

下载镜像

开始 -> 运行 -> powershell

找到镜像的名称,在docker hub上查看。这里使用了一个最小版本的镜像标签。
需要注意的是,文档上的镜像标签不是最新的,要以上面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

猜你喜欢

转载自blog.csdn.net/qq_25439417/article/details/84285126