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:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    8000/tcp
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_TAO_TOOLKIT_VERSION=6.25.7-PyTorch
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=6.25.7
    10/23/2025 8:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_PRODUCT_NAME=TAO Toolkit
    10/23/2025 8:56 PM UTC
    sha256:bdcc95a7c18de278f332ccb2cd13d1de4f4503ac8470dfd6229c1590e2c8c651RUN
    GDRCOPY_VERSION=v2.4.4 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 NCCL_VERSION=2.25.1-1 PYTHON_VERSION=3.12 pip uninstall -y ray
    10/23/2025 8:56 PM UTC
    sha256:06112e41af421e3f1f058ecaac0849bb9d64fe55b603d32417c339a1f5592c03RUN
    GDRCOPY_VERSION=v2.4.4 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 NCCL_VERSION=2.25.1-1 PYTHON_VERSION=3.12 . /opt/venv/cosmos_rl/bin/activate &&
      cd tao-core &&
      bash release/python/build_wheel.sh &&
      find dist/ -name "nvidia_tao_core*.whl" -type f | xargs -n 1 pip install &&
      cp nvidia_tao_core/microservices/nginx.conf /etc/nginx/ &&
      cd .. &&
      rm -rf tao-core
    10/23/2025 8:56 PM UTC
    ...