NVIDIA
NVIDIA
TAO Toolkit
Container
NVIDIA
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    ["/bin/sh" "-c" "if [ \"$RUN_CLI\" = \"1\" ]; then         /bin/bash;     else         /bin/bash $(get-microservice-script);     fi"]
    07/31/2026 8:55 PM UTC
    sha256:2e7aebc14806e4ed3cbc170f6edc4a4b73d45c96b84597fb6b378cf1b101200eRUN
    cd /opt/nvidia/wheels &&
      ls ./*.whl | xargs -I'{}' python -m pip install '{}' &&
      rm -f *.whl &&
      ARCH_LIB_PATH="/usr/lib/$(uname -m)-linux-gnu" &&
      sed -i "s|export LD_LIBRARY_PATH=${ARCH_LIB_PATH}:\${LD_LIBRARY_PATH}|export LD_LIBRARY_PATH=\${LD_LIBRARY_PATH}:${ARCH_LIB_PATH}|g" /etc/bash.bashrc &&
      { find / -type d -name ".git" -exec rm -rf {} + 2>/dev/null; find / -type f -name ".git" -exec rm -f {} + 2>/dev/null; find / -type f -name ".gitignore" -exec rm -f {} + 2>/dev/null; find / -type d -name ".github" -exec rm -rf {} + 2>/dev/null; find / -type d -name ".gitlab" -exec rm -rf {} + 2>/dev/null; find / -type f -name ".gitlab-ci.yml" -exec rm -f {} + 2>/dev/null; find / -type f -name "gitlab-ci.yml" -exec rm -f {} + 2>/dev/null; true; }
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    8000/tcp
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    CUBLAS_WORKSPACE_CONFIG=:4096:8
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_TAO_TOOLKIT_VERSION=7.1.0-PyTorch
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=7.1.0
    07/31/2026 8:55 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_PRODUCT_NAME=TAO Toolkit
    07/31/2026 8:55 PM UTC