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=76e7657e2f0914c2996c77f55afb3834bc5faf24
    04/28/2021 8:05 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    04/28/2021 8:05 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=22596402
    04/28/2021 8:05 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=22596402
    04/28/2021 8:05 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    04/28/2021 8:05 PM UTC
    sha256:1f924a0df31b2d8c004832e20fd977dd60adc039f76fdd7a09613d2d020df874RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.05-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
    04/28/2021 8:05 PM UTC
    sha256:c51725cc81a0fbfb8e7567f60ed27c700e4a404bd9cd3ec3007b4afbef729504RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.05-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
    04/28/2021 8:05 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
    04/28/2021 8:05 PM UTC
    sha256:629e0c450a3085538748c53b667586593a73100acde5d8dc5d5235887a9e34c4RUN
    TENSORFLOW_VERSION=2.4.0 NVIDIA_TENSORFLOW_VERSION=21.05-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
    04/28/2021 8:05 PM UTC
    sha256:1d6bfb8fe0304e35605abc15898294a54cd0bf21fe40eac9fc7e03235114b1f4COPY
    nvidia-examples /workspace/nvidia-examples
    04/28/2021 8:05 PM UTC
    ...