# Expecting base image to be the image built by ../cu112/Dockerfile.ec2.gpu
ARG BASE_IMAGE=""

FROM $BASE_IMAGE

# Add any script or repo as required
RUN git clone -b v0.1 https://github.com/aws-samples/deep-learning-models
WORKDIR "/"