update dockerfile according to the new INSTALL.md (#883)

From https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/INSTALL.md:
1. additionally install tqdm, cudatoolkit, apex
2. add FORCE_CUDA argument

docker/docker-jupyter/Dockerfile

@@ -28,11 +28,14 @@ ENV PATH=$CONDA_PREFIX/bin:$PATH
 ENV CONDA_AUTO_UPDATE_CONDA=false
 
 RUN conda install -y ipython
-RUN pip install requests ninja yacs cython matplotlib jupyter
+RUN pip install requests ninja yacs cython matplotlib jupyter tqdm
 
-# Install PyTorch 1.0 Nightly and OpenCV
-RUN conda install -y pytorch-nightly -c pytorch \
- && conda install -y opencv -c menpo \
+# Install PyTorch Nightly
+ARG CUDA
+RUN conda install -y pytorch-nightly cudatoolkit=${CUDA} -c pytorch
+
+# Install OpenCV
+RUN conda install -y opencv -c menpo \
  && conda clean -ya
 
 WORKDIR /root
@@ -53,7 +56,14 @@ RUN git clone https://github.com/cocodataset/cocoapi.git \
  && cd cocoapi/PythonAPI \
  && python setup.py build_ext install
 
+# install apex
+RUN git clone https://github.com/NVIDIA/apex.git \
+ && cd apex \
+ && python setup.py install --cuda_ext --cpp_ext
+
 # install PyTorch Detection
+ARG FORCE_CUDA="1"
+ENV FORCE_CUDA=${FORCE_CUDA}
 RUN git clone https://github.com/facebookresearch/maskrcnn-benchmark.git \
  && cd maskrcnn-benchmark \
  && python setup.py build develop