syliva/mmdetection3d repository overview

⁠basic envs

ARG PYTORCH="1.6.0" ARG CUDA="10.1" ARG CUDNN="7" ARG MMCV="1.6.0" ARG MMDET="2.24.0" ARG MMSEG="0.20.0" FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

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))/../"

⁠To fix GPG key error when running apt-get update

RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub⁠ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub⁠

⁠change ubuntu sources to tsinghua

RUN sed -i 's#http://archive.ubuntu.com/#http://mirrors.tuna.tsinghua.edu.cn/#⁠' /etc/apt/sources.list; RUN apt-get update --fix-missing
&& apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 --fix-missing
&& apt-get clean
&& rm -rf /var/lib/apt/lists/*

⁠Install MMCV, MMDetection and MMSegmentation

ARG PYTORCH ARG CUDA ARG MMCV ARG MMDET ARG MMSEG

⁠change pip sources to aliyun

RUN pip install -i http://mirrors.aliyun.com/pypi/simple⁠ -U pip
&& pip config set global.index-url http://mirrors.aliyun.com/pypi/simple⁠
&& pip config set install.trusted-host mirrors.aliyun.com RUN ["/bin/bash", "-c", "pip install --no-cache-dir mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html⁠"] RUN pip install --no-cache-dir mmdet==${MMDET} mmsegmentation==${MMSEG}

⁠Install MMDetection3D

RUN conda clean --all COPY . /mmdetection3d WORKDIR /mmdetection3d ENV FORCE_CUDA="1" RUN pip install -r requirements/build.txt RUN pip install --no-cache-dir -e .