diff --git a/docker/Dockerfile b/docker/Dockerfile index 762441fe2..95ff521a2 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -2,7 +2,6 @@ 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 @@ -11,32 +10,37 @@ RUN apt-get update -y \ && apt-get install -y libglib2.0-0 libsm6 libxext6 libxrender-dev # 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="/root/miniconda3/bin:${PATH}" +ARG PATH="/root/miniconda3/bin:${PATH}" +RUN apt-get update + +RUN apt-get install -y wget && rm -rf /var/lib/apt/lists/* -ENV PATH=/miniconda/bin:$PATH +RUN wget \ + https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \ + && mkdir /root/.conda \ + && bash Miniconda3-latest-Linux-x86_64.sh -b \ + && rm -f Miniconda3-latest-Linux-x86_64.sh H # 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 +RUN /root/miniconda3/bin/conda install -y conda-build \ + && /root/miniconda3/bin/conda create -y --name py36 python=3.6.7 \ + && /root/miniconda3/bin/conda clean -ya ENV CONDA_DEFAULT_ENV=py36 -ENV CONDA_PREFIX=/miniconda/envs/$CONDA_DEFAULT_ENV +ENV CONDA_PREFIX=/root/miniconda3/envs/$CONDA_DEFAULT_ENV 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 opencv-python tqdm +RUN pip install ninja yacs cython matplotlib # Install PyTorch 1.0 Nightly -ARG CUDA -RUN conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch \ - && conda clean -ya +RUN conda install -y pytorch-nightly -c pytorch && conda clean -ya + # Install TorchVision master +RUN pip install --upgrade torch RUN git clone https://github.com/pytorch/vision.git \ && cd vision \ && python setup.py install @@ -46,16 +50,7 @@ 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 +RUN git clone https://github.com/facebookresearch/maskrcnn-benchmark.git WORKDIR /maskrcnn-benchmark