DockerDependencyFile

# Setup required args
ARG PETSC_BASE_IMAGE=ghcr.io/ubchrest/petsc-docker/petsc-build-gcc:latest
ARG TENSORFLOW_IMAGE=ghcr.io/ubchrest/tensorflow-docker/tensorflow-build:latest

# Setup pull from tensorflow as workaround for --from not seeing argument
FROM $TENSORFLOW_IMAGE as tensor_flow_image

# Use the petsc-build as the base
ARG PETSC_BASE_IMAGE
FROM $PETSC_BASE_IMAGE

# copy the tensorflow image
COPY --from=tensor_flow_image /tensorflow /tensorflow
ENV TENSORFLOW_DIR=/tensorflow