ARG FROM_IMAGE_NAME=nvcr.io/nvidia/mxnet:20.12-py3

FROM $FROM_IMAGE_NAME

WORKDIR /workspace/rn50

COPY requirements.txt .
RUN pip install -r requirements.txt

COPY . .