--- a
+++ b/docker/Dockerfile
@@ -0,0 +1,52 @@
+FROM pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-devel
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    libsm6 \
+    libxext6 \
+    libfontconfig1 \
+    libxrender1 \
+    libswscale-dev \
+    libtbb2 \
+    libtbb-dev \
+    libjpeg-dev \
+    libpng-dev \
+    libtiff-dev \
+    libjasper-dev \
+    libavformat-dev \
+    libpq-dev \
+    libturbojpeg \
+    software-properties-common \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+RUN pip install catalyst
+RUN pip install albumentations
+RUN pip install cnn_finetune
+RUN pip install timm
+RUN pip install click
+RUN pip install pandas
+RUN pip install SimpleITK
+RUN pip install segmentation-models-pytorch
+RUN pip install wandb
+RUN pip install pyarrow
+RUN pip install lz4
+RUN pip install pydicom
+RUN pip install iterative-stratification
+
+# Install mmdetection
+RUN pip install Cython
+RUN git clone https://github.com/open-mmlab/mmdetection.git /mmdetection
+WORKDIR /mmdetection
+RUN pip install --no-cache-dir -e .
+
+WORKDIR /tmp/unique_for_apex
+# uninstall Apex if present, twice to make absolutely sure :)
+RUN pip uninstall -y apex || :
+RUN pip uninstall -y apex || :
+# SHA is something the user can touch to force recreation of this Docker layer,
+# and therefore force cloning of the latest version of Apex
+RUN SHA=ToUcHMe git clone https://github.com/NVIDIA/apex.git
+WORKDIR /tmp/unique_for_apex/apex
+RUN pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .
+WORKDIR /workspace