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