From 65da3e37521c7fe366efd6ac55a73db6ee404065 Mon Sep 17 00:00:00 2001 From: Jerry Jiarui XU Date: Tue, 29 Sep 2020 19:54:32 +0800 Subject: [PATCH] update docker (#149) * update docker * Update Dockerfile --- README.md | 4 +++- docker/Dockerfile | 7 ++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 16fd84c44..7a929d132 100644 --- a/README.md +++ b/README.md @@ -8,6 +8,8 @@ [![badge](https://github.com/open-mmlab/mmsegmentation/workflows/build/badge.svg)](https://github.com/open-mmlab/mmsegmentation/actions) [![codecov](https://codecov.io/gh/open-mmlab/mmsegmentation/branch/master/graph/badge.svg)](https://codecov.io/gh/open-mmlab/mmsegmentation) [![license](https://img.shields.io/github/license/open-mmlab/mmsegmentation.svg)](https://github.com/open-mmlab/mmsegmentation/blob/master/LICENSE) +[![issue resolution](https://isitmaintained.com/badge/resolution/open-mmlab/mmsegmentation.svg)](https://github.com/open-mmlab/mmsegmentation/issues) +[![open issues](https://isitmaintained.com/badge/open/open-mmlab/mmsegmentation.svg)](https://github.com/open-mmlab/mmsegmentation/issues) Documentation: https://mmsegmentation.readthedocs.io/ @@ -16,7 +18,7 @@ Documentation: https://mmsegmentation.readthedocs.io/ MMSegmentation is an open source semantic segmentation toolbox based on PyTorch. It is a part of the OpenMMLab project. -The master branch works with **PyTorch 1.3 to 1.5**. +The master branch works with **PyTorch 1.3 to 1.6**. ![demo image](resources/seg_demo.gif) diff --git a/docker/Dockerfile b/docker/Dockerfile index 2612f6fb9..8e090f73a 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -ARG PYTORCH="1.5" +ARG PYTORCH="1.6.0" ARG CUDA="10.1" ARG CUDNN="7" @@ -8,13 +8,14 @@ ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0+PTX" ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all" ENV CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" -RUN apt-get update && apt-get install -y libglib2.0-0 libsm6 libxrender-dev libxext6 \ +RUN apt-get update && apt-get install -y git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* # Install mmsegmentation RUN conda clean --all -RUN pip install mmcv-full==latest+torch1.5.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html + +RUN pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html RUN git clone https://github.com/open-mmlab/mmsegmenation.git /mmsegmentation WORKDIR /mmsegmentation RUN pip install -r requirements/build.txt