segmentation_models_pytorch/docker/Dockerfile

FROM anibali/pytorch:cuda-9.0

RUN pip install segmentation-models-pytorch