Add requirements for running pycaffe in the development docker image.

BVLC · Feb 11, 2016 · ff0ca8e · ff0ca8e
1 parent e2a1ced
commit ff0ca8e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/docker/devel/Dockerfile b/docker/devel/Dockerfile
@@ -4,21 +4,30 @@ MAINTAINER team-sl@zalando.de
 RUN apt-get update \
  && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     cmake \
+    cmake-curses-gui \
     git \
+    ca-certificates \
+    git \
+    bc \
+    build-essential \
     libatlas-base-dev \
     libatlas-dev \
     libboost-all-dev \
     libgflags-dev \
     libgoogle-glog-dev \
-    libhdf5-serial-dev \
+    libhdf5-dev \
     libleveldb-dev \
     liblmdb-dev \
     libopenblas-dev \
     libopencv-dev \
     libprotobuf-dev \
     libsnappy-dev \
     protobuf-compiler \
+    python-dev \
     python-numpy \
+    python-pip \
+    python-protobuf \
+    python-skimage \
  && apt-get clean \
  && rm -rf /var/lib/apt/lists/*
 

diff --git a/docker/start_caffe_docker.sh b/docker/start_caffe_docker.sh
@@ -26,5 +26,6 @@ eval ${DOCKER} run \
     -h $DOCKER_HOST \
     -v $CAFFE_ROOT:$CAFFE_ROOT \
     -w $CAFFE_ROOT \
+    -e PYTHONPATH=$CAFFE_ROOT/python \
     -u $USER_ID:$GROUP_ID \
     $DOCKER_IMAGE bash