Skip to main content
Google
TensorFlow
Container
Google
TensorFlow

TensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.ref=338d6d65917bd20740ff3fb4ec7551ec58f9a3e8
    11/11/2021 7:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=29194758
    11/11/2021 7:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=29194758
    11/11/2021 7:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    11/11/2021 7:35 PM UTC
    sha256:0f324ffd7b54aecb05e9f0b85f8562a14355b9f457ecb990d69f6c4803d6a157RUN
    TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.11-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=29194758 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=f31681fa22fb23401a8c6c8973cb025aef135272 NVIDIA_BUILD_REF=338d6d65917bd20740ff3fb4ec7551ec58f9a3e8 ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib &&
      echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf &&
      ldconfig &&
      rm -f ${_CUDA_COMPAT_PATH}/lib
    11/11/2021 7:35 PM UTC
    sha256:f0d1d5d03cf235713f8a6738ba9a8af180e9a09565a2c59f4dd6d49e0116ea15RUN
    TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.11-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=29194758 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=f31681fa22fb23401a8c6c8973cb025aef135272 NVIDIA_BUILD_REF=338d6d65917bd20740ff3fb4ec7551ec58f9a3e8 pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy==0.33.0
    11/11/2021 7:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
    11/11/2021 7:35 PM UTC
    sha256:4e264e1c59c78ce944a6a2bff5a74daf52d1986c3d995190dd85dfa47662995fRUN
    TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.11-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=29194758 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=f31681fa22fb23401a8c6c8973cb025aef135272 NVIDIA_BUILD_REF=338d6d65917bd20740ff3fb4ec7551ec58f9a3e8 URL=$(VERIFY=1 /nvidia/build-scripts/installTRT.sh 2>/dev/null | sed -n "s/^.*\(http.*\)deb.*$/\1/p")tar &&
      FILE=$(wget -O - $URL 2>/dev/null | sed -n 's/^.*href="\(TensorRT[^"]*\)".*$/\1/p' | grep -v internal) &&
      wget --quiet $URL/$FILE -O - | tar -xz &&
      PY=$(python -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') &&
      pip install TensorRT-*/python/tensorrt-*-cp$PY*.whl &&
      pip install TensorRT-*/graphsurgeon/graphsurgeon-*.whl &&
      pip install TensorRT-*/uff/uff-*.whl &&
      mv /usr/src/tensorrt /opt &&
      ln -s /opt/tensorrt /usr/src/tensorrt &&
      rm -r TensorRT-* &&
      UFF_PATH=$(pip show uff | sed -n 's/Location: \(.*\)$/\1/p')/uff &&
      sed -i 's/from tensorflow import GraphDef/from tensorflow.python import GraphDef/' $UFF_PATH/converters/tensorflow/conversion_helpers.py &&
      chmod +x ${UFF_PATH}/bin/convert_to_uff.py &&
      ln -sf ${UFF_PATH}/bin/convert_to_uff.py /usr/local/bin/convert-to-uff
    11/11/2021 7:35 PM UTC
    sha256:495469671573a464329d3b62a41b272fa2811a8a04d0434f7f201fa208da4659COPY
    nvidia-examples /workspace/nvidia-examples
    11/11/2021 7:34 PM UTC
    sha256:d1dc150b145b0035813fa3aa110638a93d557b4ec8caf62a941304309dfb9001RUN
    TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.11-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=29194758 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=f31681fa22fb23401a8c6c8973cb025aef135272 NVIDIA_BUILD_REF=338d6d65917bd20740ff3fb4ec7551ec58f9a3e8 sed -i "s/NVIDIA_TENSORFLOW_VERSION/$NVIDIA_TENSORFLOW_VERSION/g" docker-examples/*
    11/11/2021 7:34 PM UTC
    ...

    NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.