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=010aae694ba2e8ed19fcd798bc36b1115181cdec
09/22/2022 9:39 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=44878075
09/22/2022 9:39 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=44878075
09/22/2022 9:39 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
09/22/2022 9:39 PM UTC
sha256:7af57506c5e290d3d6f077cfabd28cb58a3644af77bcbee0621081885744af70RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.9.1 NVIDIA_TENSORFLOW_VERSION=22.09-tf2 PYVER=3.8 BAZEL_VERSION=5.0.0 TFAPI=2 BAZEL_CACHE= NVIDIA_BUILD_REF=010aae694ba2e8ed19fcd798bc36b1115181cdec 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
09/22/2022 9:39 PM UTC
sha256:4850c542a2a8320890bc54a2da7aebef5e77fd40e5a8bab32be223f5ed926d24RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.9.1 NVIDIA_TENSORFLOW_VERSION=22.09-tf2 PYVER=3.8 BAZEL_VERSION=5.0.0 TFAPI=2 BAZEL_CACHE= NVIDIA_BUILD_REF=010aae694ba2e8ed19fcd798bc36b1115181cdec pip install --no-cache-dir git+https://github.com/kaixih/tf_op_graph.git@v0.1
09/22/2022 9:39 PM UTC
sha256:981f04f6643028f319b16ca119516131b2d09fdc1b01f51f6bda27f88df8e21aRUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.9.1 NVIDIA_TENSORFLOW_VERSION=22.09-tf2 PYVER=3.8 BAZEL_VERSION=5.0.0 TFAPI=2 BAZEL_CACHE= NVIDIA_BUILD_REF=010aae694ba2e8ed19fcd798bc36b1115181cdec pip install pydot &&
  apt-get update &&
  apt-get install -y graphviz
09/22/2022 9:39 PM UTC
sha256:8fdac5be20b319392cf34e926bca6337803f60def6d493fdb6b5299a922b90f1RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.9.1 NVIDIA_TENSORFLOW_VERSION=22.09-tf2 PYVER=3.8 BAZEL_VERSION=5.0.0 TFAPI=2 BAZEL_CACHE= NVIDIA_BUILD_REF=010aae694ba2e8ed19fcd798bc36b1115181cdec 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==${POLYGRAPHY_VERSION}
09/22/2022 9:38 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
09/22/2022 9:38 PM UTC
sha256:9b471b1082459ec3ae72b5af84979a1d04fcde3fdfe66b4d4260bda9aa255077RUN
TARGETARCH=amd64 TENSORFLOW_VERSION=2.9.1 NVIDIA_TENSORFLOW_VERSION=22.09-tf2 PYVER=3.8 BAZEL_VERSION=5.0.0 TFAPI=2 BAZEL_CACHE= NVIDIA_BUILD_REF=010aae694ba2e8ed19fcd798bc36b1115181cdec URL=$(VERIFY=1 /nvidia/build-scripts/installTRT.sh 2>/dev/null | sed -n "s/^.*\(http.*\)tar.*$/\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
09/22/2022 9:38 PM UTC
1-10 of 121 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.