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=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
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.