ARG BASE_IMAGE=onnxtraining.azurecr.io/azureml/base-gpu:openmpi4.0.0-cuda10.1-cudnn7-ubuntu16.04
FROM ${BASE_IMAGE}

WORKDIR /workspace

COPY onnxruntime_training_bert /workspace