Skip to main content
NVIDIA
TAO Toolkit
Container
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"]
    08/31/2026 11:26 PM UTC
    sha256:3a3084aaa78edd99839c247b2c2e8355953bde44435857e68139776bb8da2f31RUN
    cd /opt/nvidia/wheels &&
      ls ./*.whl | xargs -I'{}' python -m pip install '{}' &&
      rm -f *.whl &&
      python -m pip install 'av==17.1.0' 'onnxscript==0.7.1' &&
      python -c 'import importlib.util; from importlib.metadata import version; import av, onnxscript; assert version("av") == "17.1.0"; assert version("onnxscript") == "0.7.1"; assert importlib.util.find_spec("decord") is None; print(f"FC video dependencies: av={av.__version__}, onnxscript={onnxscript.__version__}, decord=absent")' &&
      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; }
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    8000/tcp
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    CUBLAS_WORKSPACE_CONFIG=:4096:8
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_TAO_TOOLKIT_VERSION=7.2.0-PyTorch
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=7.2.0
    08/31/2026 11:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_PRODUCT_NAME=TAO Toolkit
    08/31/2026 11:26 PM UTC