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"]
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    8000/tcp
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    CUBLAS_WORKSPACE_CONFIG=:4096:8
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_TAO_TOOLKIT_VERSION=6.25.9-PyTorch
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=6.25.9
    09/18/2025 7:56 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_PRODUCT_NAME=TAO Toolkit
    09/18/2025 7:56 AM UTC
    sha256:c7d3368d3f171461f9c22ff9c605cd5393a050905a9f07a9ef7750366c6d2709RUN
    cd wheels &&
      ls ./*.whl|xargs -I'{}' python -m pip install '{}' &&
      rm *.whl
    09/18/2025 7:56 AM UTC
    ...