FROM nvcr.io/nvidia/pytorch:21.06-py3 RUN rm -rf /workspace/* WORKDIR /workspace/unet ADD requirements.txt . RUN pip install --no-cache-dir --upgrade --pre pip RUN pip install --no-cache-dir -r requirements.txt ADD . .