FROM anibali/pytorch:cuda-9.0

RUN pip install segmentation-models-pytorch