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"]
    05/27/2026 1:00 AM UTC
    sha256:3bb405c508098ae3aca9d0391295ce87adb9150a3d20568b44775dddd16e42c9RUN
    TARGETARCH=amd64 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 COSMOS_RL_GITHUB_REPO=https://gitlab-master.nvidia.com/nvidia-tao-toolkit/cosmos-reason.git COSMOS_RL_GITHUB_BRANCH=feature/enhanced-hooks-and-custom-loggers DEEPEP_COMMIT=567632d USE_LOCAL_COSMOS_RL_GITHUB=false LOCAL_COSMOS_RL_PATH=. LOCAL_TAO_CORE_PATH=tao-core LOCAL_COSMOS_RL_GITHUB_PATH=cosmos-rl-github LOCAL_TAO_DAFT_PATH=nvidia-tao-daft COSMOS_RL_EXTRAS=tao find / -type d -name ".git" -exec rm -rf {} + 2>/dev/null ||
      true &&
      find / -type f -name ".git" -exec rm -f {} + 2>/dev/null ||
      true &&
      find / -type f -name ".gitignore" -exec rm -f {} + 2>/dev/null ||
      true &&
      find / -type f -name ".gitmodules" -exec rm -f {} + 2>/dev/null ||
      true &&
      find / -type d -name ".github" -exec rm -rf {} + 2>/dev/null ||
      true &&
      find / -type d -name ".gitlab" -exec rm -rf {} + 2>/dev/null ||
      true &&
      find / -type f -name ".gitlab-ci.yml" -exec rm -f {} + 2>/dev/null ||
      true &&
      find / -type f -name "gitlab-ci.yml" -exec rm -f {} + 2>/dev/null ||
      true
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    [8000/tcp]
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_TAO_TOOLKIT_VERSION=6.25.11-PyTorch
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=6.25.11
    05/27/2026 1:00 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_PRODUCT_NAME=TAO Toolkit
    05/27/2026 1:00 AM UTC
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
    TARGETARCH=amd64 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 COSMOS_RL_GITHUB_REPO=https://gitlab-master.nvidia.com/nvidia-tao-toolkit/cosmos-reason.git COSMOS_RL_GITHUB_BRANCH=feature/enhanced-hooks-and-custom-loggers DEEPEP_COMMIT=567632d USE_LOCAL_COSMOS_RL_GITHUB=false LOCAL_COSMOS_RL_PATH=. LOCAL_TAO_CORE_PATH=tao-core LOCAL_COSMOS_RL_GITHUB_PATH=cosmos-rl-github LOCAL_TAO_DAFT_PATH=nvidia-tao-daft COSMOS_RL_EXTRAS=tao pip uninstall -y xformers
    05/27/2026 1:00 AM UTC
    ...