| # 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=18.04 |
| |
| ARG ARCH= |
| ARG CUDA=10.1 |
| FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-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=7.6.4.38-1 |
| ARG CUDNN_MAJOR_VERSION=7 |
| ARG LIB_DIR_PREFIX=x86_64 |
| ARG LIBNVINFER=6.0.1-1 |
| ARG LIBNVINFER_MAJOR_VERSION=6 |
| |
| # Needed for string substitution |
| SHELL ["/bin/bash", "-c"] |
| RUN apt-get update && apt-get install -y --no-install-recommends \ |
| build-essential \ |
| cuda-command-line-tools-${CUDA/./-} \ |
| # There appears to be a regression in libcublas10=10.2.2.89-1 which |
| # prevents cublas from initializing in TF. See |
| # https://github.com/tensorflow/tensorflow/issues/9489#issuecomment-562394257 |
| libcublas10=10.2.1.243-1 \ |
| libcublas-dev=10.2.1.243-1 \ |
| cuda-nvrtc-${CUDA/./-} \ |
| cuda-nvrtc-dev-${CUDA/./-} \ |
| cuda-cudart-dev-${CUDA/./-} \ |
| cuda-cufft-dev-${CUDA/./-} \ |
| cuda-curand-dev-${CUDA/./-} \ |
| cuda-cusolver-dev-${CUDA/./-} \ |
| cuda-cusparse-dev-${CUDA/./-} \ |
| libcudnn7=${CUDNN}+cuda${CUDA} \ |
| libcudnn7-dev=${CUDNN}+cuda${CUDA} \ |
| libcurl3-dev \ |
| libfreetype6-dev \ |
| libhdf5-serial-dev \ |
| libzmq3-dev \ |
| pkg-config \ |
| rsync \ |
| software-properties-common \ |
| unzip \ |
| zip \ |
| zlib1g-dev \ |
| wget \ |
| git \ |
| && \ |
| find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ |
| rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a |
| |
| # Install TensorRT if not building for PowerPC |
| RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \ |
| apt-get install -y --no-install-recommends libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \ |
| libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \ |
| libnvinfer-plugin-dev=${LIBNVINFER}+cuda${CUDA} \ |
| libnvinfer-plugin${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \ |
| && apt-get clean \ |
| && rm -rf /var/lib/apt/lists/*; } |
| |
| # Configure the build for our CUDA configuration. |
| ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs:/usr/include/x64_64-linux-gnu:$LD_LIBRARY_PATH |
| ENV TF_NEED_CUDA 1 |
| ENV TF_NEED_TENSORRT 1 |
| ENV TF_CUDA_VERSION=${CUDA} |
| ENV TF_CUDNN_VERSION=${CUDNN_MAJOR_VERSION} |
| # 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 |
| RUN test "${CHECKOUT_TF_SRC}" -eq 1 && git clone https://github.com/tensorflow/tensorflow.git /tensorflow_src || true |
| |
| # 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 |
| |
| 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 |
| |
| RUN apt-get update && apt-get install -y \ |
| build-essential \ |
| curl \ |
| git \ |
| openjdk-8-jdk \ |
| ${PYTHON}-dev \ |
| virtualenv \ |
| swig |
| |
| RUN ${PIP} --no-cache-dir install \ |
| Pillow \ |
| h5py \ |
| keras_preprocessing \ |
| matplotlib \ |
| mock \ |
| numpy \ |
| scipy \ |
| sklearn \ |
| pandas \ |
| portpicker \ |
| && test "${USE_PYTHON_3_NOT_2}" -eq 1 && true || ${PIP} --no-cache-dir install \ |
| enum34 |
| |
| # Build and install bazel |
| ENV BAZEL_VERSION 0.15.0 |
| WORKDIR / |
| RUN mkdir /bazel && \ |
| cd /bazel && \ |
| curl -fSsL -O https://github.com/bazelbuild/bazel/releases/download/$BAZEL_VERSION/bazel-$BAZEL_VERSION-dist.zip && \ |
| unzip bazel-$BAZEL_VERSION-dist.zip && \ |
| bash ./compile.sh && \ |
| cp output/bazel /usr/local/bin/ && \ |
| rm -rf /bazel && \ |
| cd - |
| |
| COPY bashrc /etc/bash.bashrc |
| RUN chmod a+rwx /etc/bash.bashrc |
| |
| RUN ${PIP} install jupyter matplotlib |
| # https://github.com/ipython/ipykernel/issues/422 |
| RUN test "${USE_PYTHON_3_NOT_2}" == "1" && ${PIP} install ipykernel==5.1.1 nbformat==4.4.0 || true |
| RUN ${PIP} install jupyter_http_over_ws |
| 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 install -y --no-install-recommends wget |
| 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 ${PYTHON} -m ipykernel.kernelspec |
| |
| CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root"] |