caffe2環境搭建(windows 7 + docker)
阿新 • • 發佈:2018-11-20
背景
背景不多說,想體驗一下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"]}
這裡需要使用者自己註冊,然後使用系統分配字首,最終如下: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