content: # Copyright 2019 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=20.04
  
  FROM ubuntu:${UBUNTU_VERSION} AS base
  
  ARG DEBIAN_FRONTEND="noninteractive"
  
  RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
          build-essential \
          curl \
          git \
          libcurl3-dev \
          libfreetype6-dev \
          libhdf5-serial-dev \
          libzmq3-dev \
          pkg-config \
          rsync \
          software-properties-common \
          sudo \
          unzip \
          zip \
          zlib1g-dev \
          openjdk-8-jdk \
          openjdk-8-jre-headless \
          && \
      apt-get clean && \
      rm -rf /var/lib/apt/lists/*
  
  ENV CI_BUILD_PYTHON python
  
  # CACHE_STOP is used to rerun future commands, otherwise cloning tensorflow will be cached and will not pull the most recent version
  ARG CACHE_STOP=1
  # Check out TensorFlow source code if --build-arg CHECKOUT_TF_SRC=1
  ARG CHECKOUT_TF_SRC=0
  ARG TF_BRANCH=master
  RUN test "${CHECKOUT_TF_SRC}" -eq 1 && git clone https://github.com/tensorflow/tensorflow.git --branch "${TF_BRANCH}" --single-branch /tensorflow_src || true
  
  # See http://bugs.python.org/issue19846
  ENV LANG C.UTF-8
  ARG PYTHON=python3
  
  RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
      ${PYTHON} \
      ${PYTHON}-pip
  RUN ${PYTHON} -m pip --no-cache-dir install --upgrade \
      pip \
      setuptools
  
  # Some TF tools expect a "python" binary
  RUN ln -s $(which ${PYTHON}) /usr/local/bin/python
  
  RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
      curl
  
  # Install bazel
  ARG BAZEL_VERSION=3.7.2
  RUN mkdir /bazel && \
      curl -fSsL -o /bazel/installer.sh "https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh" && \
      curl -fSsL -o /bazel/LICENSE.txt "https://raw.githubusercontent.com/bazelbuild/bazel/master/LICENSE" && \
      bash /bazel/installer.sh && \
      rm -f /bazel/installer.sh
  
  ARG DEBIAN_FRONTEND="noninteractive"
  
  # install libnuma, openssh, wget
  RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
      libopenmpi-dev \
      openmpi-bin \
      openmpi-common \
      openssh-client \
      openssh-server && \
      apt-get clean && \
      rm -rf /var/lib/apt/lists/*
  
  # Create a wrapper for OpenMPI to allow running as root by default
  RUN mv /usr/bin/mpirun /usr/bin/mpirun.real && \
      echo '#!/bin/bash' > /usr/bin/mpirun && \
      echo 'mpirun.real --allow-run-as-root "$@"' >> /usr/bin/mpirun && \
      chmod a+x /usr/bin/mpirun
  
  # Configure OpenMPI to run good defaults:
  RUN echo "btl_tcp_if_exclude = lo,docker0" >> /etc/openmpi/openmpi-mca-params.conf
  
  # Install OpenSSH for MPI to communicate between containers
  RUN mkdir -p /var/run/sshd
  
  # Allow OpenSSH to talk to containers without asking for confirmation
  RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_config.new && \
      echo "    StrictHostKeyChecking no" >> /etc/ssh/ssh_config.new && \
      mv /etc/ssh/ssh_config.new /etc/ssh/ssh_config
  
  # Check out horovod source code if --build-arg CHECKOUT_HOROVOD_SRC=1
  ARG CHECKOUT_HOROVOD_SRC=0
  ARG HOROVOD_BRANCH=master
  RUN test "${CHECKOUT_HOROVOD_SRC}" -eq 1 && git clone --branch "${HOROVOD_BRANCH}" --single-branch --recursive https://github.com/uber/horovod.git /horovod_src || true
  
  COPY bashrc /etc/bash.bashrc
  RUN chmod a+rwx /etc/bash.bashrc
  