content: # Copyright 2018 The TensorFlow Authors. All Rights Reserved.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  # ============================================================================
  #
  # THIS IS A GENERATED DOCKERFILE.
  #
  # This file was assembled from multiple pieces, whose use is documented
  # throughout. Please refer to the TensorFlow dockerfiles documentation
  # for more information.
  
  ARG UBUNTU_VERSION=18.04
  
  FROM ubuntu:${UBUNTU_VERSION} as base
  
  ARG USE_PYTHON_3_NOT_2
  ARG _PY_SUFFIX=${USE_PYTHON_3_NOT_2:+3}
  ARG PYTHON=python${_PY_SUFFIX}
  ARG PIP=pip${_PY_SUFFIX}
  
  # See http://bugs.python.org/issue19846
  ENV LANG C.UTF-8
  
  RUN apt-get update && apt-get install -y \
      ${PYTHON} \
      ${PYTHON}-pip
  
  RUN ${PIP} --no-cache-dir install --upgrade \
      pip \
      setuptools
  
  # Some TF tools expect a "python" binary
  RUN ln -s $(which ${PYTHON}) /usr/local/bin/python 
  
  # Options:
  #   tensorflow
  #   tensorflow-gpu
  #   tf-nightly
  #   tf-nightly-gpu
  ARG TF_PACKAGE=tensorflow
  RUN apt-get update && apt-get install -y wget libhdf5-dev
  RUN ${PIP} install --global-option=build_ext \
              --global-option=-I/usr/include/hdf5/serial/ \
              --global-option=-L/usr/lib/powerpc64le-linux-gnu/hdf5/serial \
              h5py
  
  # CACHE_STOP is used to rerun future commands, otherwise downloading the .whl will be cached and will not pull the most recent version
  ARG CACHE_STOP=1
  RUN if [ ${TF_PACKAGE} = tensorflow-gpu ]; then \
          BASE=https://powerci.osuosl.org/job/TensorFlow_PPC64LE_GPU_Release_Build/lastSuccessfulBuild/; \
      elif [ ${TF_PACKAGE} = tf-nightly-gpu ]; then \
          BASE=https://powerci.osuosl.org/job/TensorFlow_PPC64LE_GPU_Nightly_Artifact/lastSuccessfulBuild/; \
      elif [ ${TF_PACKAGE} = tensorflow ]; then \
          BASE=https://powerci.osuosl.org/job/TensorFlow_PPC64LE_CPU_Release_Build/lastSuccessfulBuild/; \
      elif [ ${TF_PACKAGE} = tf-nightly ]; then \
          BASE=https://powerci.osuosl.org/job/TensorFlow_PPC64LE_CPU_Nightly_Artifact/lastSuccessfulBuild/; \
      fi; \
      MAJOR=`${PYTHON} -c 'import sys; print(sys.version_info[0])'`; \
      MINOR=`${PYTHON} -c 'import sys; print(sys.version_info[1])'`; \
      PACKAGE=$(wget -qO- ${BASE}"api/xml?xpath=//fileName&wrapper=artifacts" | grep -o "[^<>]*cp${MAJOR}${MINOR}[^<>]*.whl"); \
      wget ${BASE}"artifact/tensorflow_pkg/"${PACKAGE}; \
      ${PIP} install ${PACKAGE}
  
  COPY bashrc /etc/bash.bashrc
  RUN chmod a+rwx /etc/bash.bashrc
  