Skip to content
Snippets Groups Projects
Unverified Commit 225e9092 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Add Dockerfile (#1168)

* add dockerfile

* update installation guide

* minor fix
parent e6e23152
No related branches found
No related tags found
No related merge requests found
......@@ -53,7 +53,16 @@ It is recommended that you run step d each time you pull some updates from githu
2. Following the above instructions, mmdetection is installed on `dev` mode, any local modifications made to the code will take effect without the need to reinstall it (unless you submit some commits and want to update the version number).
### Prepare COCO dataset
### Another option: Docker Image
We provide a [Dockerfile](docker/Dockerfile) to build an image.
```shell
# build an image with PyTorch 1.1, CUDA 10.0 and CUDNN 7.5
docker build -t mmdetection docker/
```
### Prepare datasets
It is recommended to symlink the dataset root to `$MMDETECTION/data`.
If your folder structure is different, you may need to change the corresponding paths in config files.
......
ARG PYTORCH="1.1.0"
ARG CUDA="10.0"
ARG CUDNN="7.5"
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
RUN apt-get update && apt-get install -y libglib2.0-0 libsm6 libxrender-dev libxext6
# Install mmdetection
RUN conda install cython -y
RUN git clone https://github.com/open-mmlab/mmdetection.git /mmdetection
WORKDIR /mmdetection
RUN pip install -e .
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment