onnxruntime/orttraining/docker/Dockerfile
2020-03-11 14:39:03 -07:00

6 lines
179 B
Docker

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