Update Dockerfile

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