quiltdata/pytorch-detectron2-demo repository overview

Docker containers used to demonstrate how to train detectron2 using Quilt to provide the data. This is a very light layer on top of the AWS Deep Learning Container (PyTorch Training, GPU). This layer just exists to make the example more compact.

FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training@sha256:8afa9bcaf77c95d54e653bf32a565ef411601d9eada255a33dd391af01d0fb09

RUN pip install quilt3 cython
RUN pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

The base container is 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.3.1-gpu-py36-cu101-ubuntu16.04, pinned to the latest version as of Jan 5th, 2019