Google
Google
TensorFlow
Container
Google
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=775322cad6c545e6ab515da1b263c3e91b12c423
    02/10/2021 9:29 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    02/10/2021 9:29 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=19902717
    02/10/2021 9:29 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=19902717
    02/10/2021 9:29 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    02/10/2021 9:29 PM UTC
    sha256:76c808aa804f37ac29e4b613aa66586d3f6c9e4fb87689785899f273c83a9734RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.02-tf2 PYVER=3.8 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 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
    02/10/2021 9:29 PM UTC
    sha256:96e8c7952ae116b3f42afb6bb89796911f1b9e0786f119c65e1943b3d16746a0RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.02-tf2 PYVER=3.8 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://tensorrt-read-only:Tensorrt\@123@urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy
    02/10/2021 9:29 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
    02/10/2021 9:29 PM UTC
    sha256:c3072206274e44291b92e105d232b8de2107cfd8212f261a76d17c4e43f8a1b6RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.02-tf2 PYVER=3.8 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 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
    02/10/2021 9:29 PM UTC
    sha256:166ee12eb5bdc75646d812396462cc5b4363af889c274f5462810b8b1071535cCOPY
    nvidia-examples /workspace/nvidia-examples
    02/10/2021 9:28 PM UTC
    ...