FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel

RUN pip install SimpleITK