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 ...@@ -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). 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`. 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. 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