Dockerfile with jupyter notebook support

INSTALL.md

@@ -65,3 +65,8 @@ Build image with defaults (`CUDA=9.0`, `CUDNN=7`):
 Build image with other CUDA and CUDNN versions:
 
     nvidia-docker build -t maskrcnn-benchmark --build-arg CUDA=9.2 --build-arg CUDNN=7 docker/ 
+
+Build and run image with built-in jupyter notebook(note that the password is used to log in jupyter notebook):
+
+    nvidia-docker build -t maskrcnn-benchmark-jupyter docker/docker-jupyter/
+    nvidia-docker run -td -p 8888:8888 -e PASSWORD=<password> -v <host-dir>:<container-dir> maskrcnn-benchmark-jupyter

docker/docker-jupyter/Dockerfile

@@ -0,0 +1,67 @@
+ARG CUDA="9.0"
+ARG CUDNN="7"
+
+FROM nvidia/cuda:${CUDA}-cudnn${CUDNN}-devel-ubuntu16.04
+
+RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections
+
+# install basics
+RUN apt-get update -y \
+ && apt-get install -y apt-utils git curl ca-certificates bzip2 cmake tree htop bmon iotop g++
+
+# Install Miniconda
+RUN curl -so /miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
+ && chmod +x /miniconda.sh \
+ && /miniconda.sh -b -p /miniconda \
+ && rm /miniconda.sh
+
+ENV PATH=/miniconda/bin:$PATH
+
+# Create a Python 3.6 environment
+RUN /miniconda/bin/conda install -y conda-build \
+ && /miniconda/bin/conda create -y --name py36 python=3.6.7 \
+ && /miniconda/bin/conda clean -ya
+
+ENV CONDA_DEFAULT_ENV=py36
+ENV CONDA_PREFIX=/miniconda/envs/$CONDA_DEFAULT_ENV
+ENV PATH=$CONDA_PREFIX/bin:$PATH
+ENV CONDA_AUTO_UPDATE_CONDA=false
+
+RUN conda install -y ipython
+RUN pip install ninja yacs cython matplotlib jupyter
+
+# Install PyTorch 1.0 Nightly and OpenCV
+RUN conda install -y pytorch-nightly -c pytorch \
+ && conda install -y opencv -c menpo \
+ && conda clean -ya
+
+WORKDIR /root
+
+USER root
+
+RUN mkdir /notebooks
+
+WORKDIR /notebooks
+
+# Install TorchVision master
+RUN git clone https://github.com/pytorch/vision.git \
+ && cd vision \
+ && python setup.py install
+
+# install pycocotools
+RUN git clone https://github.com/cocodataset/cocoapi.git \
+ && cd cocoapi/PythonAPI \
+ && python setup.py build_ext install
+
+# install PyTorch Detection
+RUN git clone https://github.com/facebookresearch/maskrcnn-benchmark.git \
+ && cd maskrcnn-benchmark \
+ && python setup.py build develop
+
+RUN jupyter notebook --generate-config
+
+ENV CONFIG_PATH="/root/.jupyter/jupyter_notebook_config.py"
+
+COPY "jupyter_notebook_config.py" ${CONFIG_PATH}
+
+ENTRYPOINT ["sh", "-c", "jupyter notebook --allow-root -y --no-browser --ip=0.0.0.0 --config=${CONFIG_PATH}"]

docker/docker-jupyter/jupyter_notebook_config.py

@@ -0,0 +1,18 @@
+import os
+from IPython.lib import passwd
+
+#c = c  # pylint:disable=undefined-variable
+c = get_config()
+c.NotebookApp.ip = '0.0.0.0'
+c.NotebookApp.port = int(os.getenv('PORT', 8888))
+c.NotebookApp.open_browser = False
+
+# sets a password if PASSWORD is set in the environment
+if 'PASSWORD' in os.environ:
+  password = os.environ['PASSWORD']
+  if password:
+    c.NotebookApp.password = passwd(password)
+  else:
+    c.NotebookApp.password = ''
+    c.NotebookApp.token = ''
+  del os.environ['PASSWORD']