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
  
  ARG ARCH=
  ARG CUDA=11.2
  FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}.1-base-ubuntu${UBUNTU_VERSION} as base
  # ARCH and CUDA are specified again because the FROM directive resets ARGs
  # (but their default value is retained if set previously)
  ARG ARCH
  ARG CUDA
  ARG CUDNN=8.1.0.77-1
  ARG CUDNN_MAJOR_VERSION=8
  ARG LIB_DIR_PREFIX=x86_64
  ARG LIBNVINFER=7.2.2-1
  ARG LIBNVINFER_MAJOR_VERSION=7
  
  # Let us install tzdata painlessly
  ENV DEBIAN_FRONTEND=noninteractive
  
  # Needed for string substitution
  SHELL ["/bin/bash", "-c"]
  # Pick up some TF dependencies
  RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \
      apt-get update && apt-get install -y --no-install-recommends \
          build-essential \
          cuda-command-line-tools-${CUDA/./-} \
          libcublas-${CUDA/./-} \
          cuda-nvrtc-${CUDA/./-} \
          libcufft-${CUDA/./-} \
          libcurand-${CUDA/./-} \
          libcusolver-${CUDA/./-} \
          libcusparse-${CUDA/./-} \
          curl \
          libcudnn8=${CUDNN}+cuda${CUDA} \
          libfreetype6-dev \
          libhdf5-serial-dev \
          libzmq3-dev \
          pkg-config \
          software-properties-common \
          unzip
  
  # Install TensorRT if not building for PowerPC
  # NOTE: libnvinfer uses cuda11.1 versions
  RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
          apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub && \
          echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /"  > /etc/apt/sources.list.d/tensorRT.list && \
          apt-get update && \
          apt-get install -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda11.0 \
          libnvinfer-plugin${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda11.0 \
          && apt-get clean \
          && rm -rf /var/lib/apt/lists/*; }
  
  # For CUDA profiling, TensorFlow requires CUPTI.
  ENV LD_LIBRARY_PATH /usr/local/cuda-11.0/targets/x86_64-linux/lib:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
  
  # Link the libcuda stub to the location where tensorflow is searching for it and reconfigure
  # dynamic linker run-time bindings
  RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 \
      && echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
      && ldconfig
  
  # See http://bugs.python.org/issue19846
  ENV LANG C.UTF-8
  
  RUN apt-get update && apt-get install -y \
      python3 \
      python3-pip
  
  RUN python3 -m pip --no-cache-dir install --upgrade \
      "pip<20.3" \
      setuptools
  
  # Some TF tools expect a "python" binary
  RUN ln -s $(which python3) /usr/local/bin/python
  
  # Options:
  #   tensorflow
  #   tensorflow-gpu
  #   tf-nightly
  #   tf-nightly-gpu
  # Set --build-arg TF_PACKAGE_VERSION=1.11.0rc0 to install a specific version.
  # Installs the latest version by default.
  ARG TF_PACKAGE=tensorflow
  ARG TF_PACKAGE_VERSION=
  RUN python3 -m pip install --no-cache-dir ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
  
  COPY bashrc /etc/bash.bashrc
  RUN chmod a+rwx /etc/bash.bashrc
  
  RUN python3 -m pip install --no-cache-dir jupyter matplotlib
  # Pin ipykernel and nbformat; see https://github.com/ipython/ipykernel/issues/422
  # Pin jedi; see https://github.com/ipython/ipython/issues/12740
  RUN python3 -m pip install --no-cache-dir jupyter_http_over_ws ipykernel==5.1.1 nbformat==4.4.0 jedi==0.17.2
  RUN jupyter serverextension enable --py jupyter_http_over_ws
  
  RUN mkdir -p /tf/tensorflow-tutorials && chmod -R a+rwx /tf/
  RUN mkdir /.local && chmod a+rwx /.local
  RUN apt-get update && apt-get install -y --no-install-recommends wget git
  WORKDIR /tf/tensorflow-tutorials
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/classification.ipynb
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/overfit_and_underfit.ipynb
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/regression.ipynb
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/save_and_load.ipynb
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/text_classification.ipynb
  RUN wget https://raw.githubusercontent.com/tensorflow/docs/master/site/en/tutorials/keras/text_classification_with_hub.ipynb
  COPY readme-for-jupyter.md README.md
  RUN apt-get autoremove -y && apt-get remove -y wget
  WORKDIR /tf
  EXPOSE 8888
  
  RUN python3 -m ipykernel.kernelspec
  
  CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root"]
  