1. 程式人生 > >caffe2環境搭建(windows 7 + docker)

caffe2環境搭建(windows 7 + docker)

背景

背景不多說,想體驗一下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