diff --git a/docker/docker-jupyter/Dockerfile b/docker/docker-jupyter/Dockerfile index bf957b636..afc512679 100644 --- a/docker/docker-jupyter/Dockerfile +++ b/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