caffe docker镜像使用及制作

1、给系统安装docker

2、下载caffe 的镜像
caffe 镜像的dockerfile,基于ubuntu16.04镜像

FROM ubuntu:16.04
LABEL maintainer caffe-maint@googlegroups.com

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential \
        cmake \
        git \
        wget \
        libatlas-base-dev \
        libboost-all-dev \
        libgflags-dev \
        libgoogle-glog-dev \
        libhdf5-serial-dev \
        libleveldb-dev \
        liblmdb-dev \
        libopencv-dev \
        libprotobuf-dev \
        libsnappy-dev \
        protobuf-compiler \
        python-dev \
        python-numpy \
        python-pip \
        python-setuptools \
        python-scipy && \
    rm -rf /var/lib/apt/lists/*

ENV CAFFE_ROOT=/opt/caffe
WORKDIR $CAFFE_ROOT

# FIXME: use ARG instead of ENV once DockerHub supports this
# https://github.com/docker/hub-feedback/issues/460
ENV CLONE_TAG=1.0

RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \
    pip install --upgrade pip && \
    cd python && for req in $(cat requirements.txt) pydot; do pip install $req; done && cd .. && \
    mkdir build && cd build && \
    cmake -DCPU_ONLY=1 .. && \
    make -j"$(nproc)"

ENV PYCAFFE_ROOT $CAFFE_ROOT/python
ENV PYTHONPATH $PYCAFFE_ROOT:$PYTHONPATH
ENV PATH $CAFFE_ROOT/build/tools:$PYCAFFE_ROOT:$PATH
RUN echo "$CAFFE_ROOT/build/lib" >> /etc/ld.so.conf.d/caffe.conf && ldconfig

WORKDIR /workspace

# 从 docker hub仓库pull caffe 的镜像
docker pull bvlc/caffe:cpu

3、运行caffe 镜像,并且启动bash

docker run -it --name test bvlc/caffe:cpu bash

docker run -it -p 7000:7000 --rm --name test docker_image_id

4、自带的caffe镜像基本上没有什么工具,因此如果需要可以安装。
首先需要设计软件源,因为caffe镜像自带的软件源是国外的,国内无法访问。

mv /etc/apt/sources.list /etc/apt/sources.list.bak
echo "deb http://mirrors.163.com/debian/ jessie main non-frzee contrib" >/etc/apt/sources.list
echo "deb http://mirrors.163.com/debian/ jessie-proposed-updates main non-free contrib" >>/etc/apt/sources.list
echo "deb-src http://mirrors.163.com/debian/ jessie main non-free contrib" >>/etc/apt/sources.list
echo "deb-src http://mirrors.163.com/debian/ jessie-proposed-updates main non-free contrib" >>/etc/apt/sources.list

#更新安装源
apt-get update 

1.修改dns 
sudo vi /etc/resolv.conf 
在里面插入: 
nameserver 8.8.8.8 
nameserver 8.8.4.4

5、安装vim

apt-get install  vim

6、安装一些必要的包,测试单独安装能够正常安装,然后在根据docker file 进行构建镜像。

pip install -U pip 
pip install -U scipy  --ignore-installed scipy
pip install -U scikit-learn 
pip install -U matplotlib 
pip install -U jupyter 
pip install -U lmdb 
pip install -U python-mnist
pip install --upgrade scikit-image

7、容器启动,以及启动jumpter notebook

#commit 以test为name的容器
docker commit test caffe_add_jumpyter:v1
#启动刚才提交的镜像,并且挂在目录,挂在的目录必须存在雨docker,以及本地目录需要使用全路径
#挂在目录  -v [宿主机目录(本地目录)]:[docker 容器目录]  不存在会自动创建
docker run -it -v /Users/gpf/Documents/tools/docker/notebooks:/caffe/notebook -p 8888:8888 --rm --name caffe_jumpyter caffe_add_jumpyter:v1 /bin/bash

# 启动jumpter notebook
jupyter notebook --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token='science'  --notebook-dir=/caffe/notebook/

8、在镜像里编译安装opencv2.4.13.6

OpenCV 官网下载源码, 建议选择2.4.13或者3.2.0版本
这里有下载好的源码,在映射的文件下
cp -r /caffe/notebook/opencv-2.4.13.6  /opt/
mkdir /opt/opencv2
cd /caffe/notebook/opencv-2.4.13.6

mkdir build

cd build

cmake -DCMAKE_INSTALL_PREFIX=/opt/opencv2/ -DCMAKE_BUILD_TYPE=Release -DBUILD_SHARED_LIBS=OFF ..

make -j8

make install

编译成功后,配置环境变量

vim ~/.zshrc (我用的是zsh)

添加
export OpenCV_DIR="/opt/opencv_2.4.13.6"
export PATH="$OpenCV_DIR/bin":$PATH
export LD_LIBRARY_PATH="$OpenCV_DIR/lib":$LD_LIBRARY_PATH
export PKG_CONFIG_PATH="$OpenCV_DIR/lib/pkgconfig:${PKG_CONFIG_PATH}"

# 将cv2.so 的路径加入pythonpath路径中,否则会出现找不到cv2 module
export PYTHONPATH=/opt/opencv2/lib/python2.7/dist-packages:$PYTHONPATH

在制作镜像时,要尽量保证镜像是轻量的,因此把不必要的文件都删除,比如opencv编译的中间文件,都删除
rm -rf   /opt/opencv-2.4.13.6/

8、docker file

FROM bvlc/caffe_revise_source:v1

RUN apt-get clean && apt-get install -y --allow-unauthenticated vim\
    && pip install -U pip\
    && pip install -U scipy  --ignore-installed scipy \
    && pip install -U scikit-learn \
    && pip install -U matplotlib \
    && pip install -U jupyter \
    && pip install -U lmdb \
    && pip install -U python-mnist \
    && pip install --upgrade scikit-image

COPY notebooks/opencv-2.4.13.6 /opt/opencv-2.4.13.6

WORKDIR /opt

RUN mkdir opencv2 \
    && cd opencv-2.4.13.6 \
    && mkdir build \
    && cd build \
    && cmake -DCMAKE_INSTALL_PREFIX=/opt/opencv2/ -DCMAKE_BUILD_TYPE=Release -DBUILD_SHARED_LIBS=OFF .. \
    && make -j8 \
    && make install \
    && rm -rf /opt/opencv-2.4.13.6/

ENV OpenCV_DIR="/opt/opencv_2.4.13.6"
ENV PATH="$OpenCV_DIR/bin":$PATH
ENV LD_LIBRARY_PATH="$OpenCV_DIR/lib":$LD_LIBRARY_PATH
ENV PKG_CONFIG_PATH="$OpenCV_DIR/lib/pkgconfig:${PKG_CONFIG_PATH}"
ENV PYTHONPATH=/opt/opencv2/lib/python2.7/dist-packages:$PYTHONPATH

# Set the working directory to /caffe
WORKDIR /caffe
VOLUME ["/caffe/data", "/caffe/notebooks"]
# Expose port 8888 (jupyter uses)
EXPOSE 8888
CMD jupyter notebook --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token='science'

注:COPY命令:是在上下文环境中的,简单来说就是需要复制的文件必须在上下文环境中,也就是在dockerfile同样的目录下,使用相对路径就可以。

9、构建docker image时,删除所有的无用的文件,与环境运行无关的文件皆可删除

#清楚pip 的缓存文件
~/.cache/pip
rm -rf  ~/.cache/pip
#
rm -rf /var/lib/apt/lists/*
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,186评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,858评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,620评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,888评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,009评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,149评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,204评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,956评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,385评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,698评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,863评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,544评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,185评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,899评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,141评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,684评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,750评论 2 351

推荐阅读更多精彩内容