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=ab18d902b6a1d880f616cdcd8c38e7bb6ca1a9c7
10/18/2021 4:49 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=28304637
10/18/2021 4:49 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=28304637
10/18/2021 4:49 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
10/18/2021 4:49 AM UTC
sha256:011faec4e5638207458647b7a08adcccf5d040f55d109c8056aa5c9d7cab8c08RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.10-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=28304637 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=5218c25b0895e0320506d21d6f4e8ed2d94a52a1 NVIDIA_BUILD_REF=ab18d902b6a1d880f616cdcd8c38e7bb6ca1a9c7 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
10/18/2021 4:49 AM UTC
sha256:49c16b8def619fc700e61b21fcc127bd195f4fc9e61a42a69cb640aa61df08edRUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.10-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=28304637 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=5218c25b0895e0320506d21d6f4e8ed2d94a52a1 NVIDIA_BUILD_REF=ab18d902b6a1d880f616cdcd8c38e7bb6ca1a9c7 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
10/18/2021 4:48 AM 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
10/18/2021 4:48 AM UTC
sha256:83f8ca3cf5b4a4eeb7b74333c216f5cb677799bae5374c66bb0c6173c8c8c104RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.6.0 NVIDIA_TENSORFLOW_VERSION=21.10-tf2 PYVER=3.8 BAZEL_VERSION=3.7.2 TFAPI=2 BAZEL_CACHE=--bazel-cache CI_JOB_ID=28304637 COMMIT_BRANCH=HEAD COMMIT_TF_SUBMODULE_SHA=5218c25b0895e0320506d21d6f4e8ed2d94a52a1 NVIDIA_BUILD_REF=ab18d902b6a1d880f616cdcd8c38e7bb6ca1a9c7 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
10/18/2021 4:48 AM UTC
sha256:178700adcb6fb63dde53bb5231df1d820c06af36fd9e269b2586bb192decee2aCOPY
nvidia-examples /workspace/nvidia-examples
10/18/2021 4:48 AM UTC
sha256:1a56b535907f0c7a9dbfcf62d0fa70b19bcfba0b932742fc0fd91a1a4d596273COPY
docker-examples docker-examples
10/18/2021 4:48 AM UTC
1-10 of 126 items

Modal Content

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.