Skip to main content
NVIDIA
TAO Toolkit
Container
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TOOLKIT_VERSION=7.1.0 NVIDIA_TAO_TOOLKIT_VERSION=7.1.0-PyTorch
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    org.opencontainers.image.created=2026-08-25T18:04:42Z org.opencontainers.image.revision=3577d0850e724e98c0de2fb87c2e83ad4940fe51 com.nvidia.tao.backend=cosmos-rl com.nvidia.tao.video-profiles=system-pyav,pynv-device-rgbp com.nvidia.tao.actions-revision=3577d0850e724e98c0de2fb87c2e83ad4940fe51
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    ACTIONS_COMMIT=3577d0850e724e98c0de2fb87c2e83ad4940fe51
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    BUILD_TIMESTAMP=2026-08-25T18:04:42Z
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    ["/bin/bash" "-lc" "if [ \"$RUN_CLI\" = \"1\" ]; then exec /bin/bash; else exec /bin/bash \"$(get-microservice-script)\"; fi"]
    08/25/2026 6:31 PM UTC
    sha256:0c64f2b86b7477a2498f6c660294fd8d45e74c6da5e5b3fc15f521b3ad974e60RUN
    TARGETARCH=amd64 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=1.17.3 PYTHON_VERSION=3.12 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 COSMOS_FRAMEWORK_REPO=https://github.com/ramanathan831/cosmos-framework.git COSMOS_FRAMEWORK_BRANCH=dev/ram/cosmos3-tao-reproducibility USE_LOCAL_COSMOS_FRAMEWORK=true LOCAL_COSMOS_FRAMEWORK_PATH=cosmos-framework-src EXPECTED_FRAMEWORK_COMMIT=03143e92a8d5273ed3c1b0d477c34af6e4eafc1e EXPECTED_FRAMEWORK_TREE=722698307100ab60837e863d3808a90b2173a917 FFMPEG_VERSION=8.1 FFMPEG_GITHUB_SHA256= NV_CODEC_HEADERS_VERSION=n12.1.14.0 PYAV_VERSION=17.1.0 PYAV_WHEEL_SHA256=f9a65d1f48b818323fb411e80358f89d77dec340b01d27c6b2dfbb9cbf4b779f VLLM_BASE_IMAGE=nvcr.io/nvidia/vllm:26.03-py3 BUILD_TIMESTAMP=2026-08-25T18:04:42Z SOURCE_DIRTY=0 ACTIONS_COMMIT=3577d0850e724e98c0de2fb87c2e83ad4940fe51 ACTIONS_TREE=46de7570e9c3ca973f7a613ebb48f4a3ec63d0aa DAFT_COMMIT=5ced3af903edfb28097e2cc18313f6c126a881eb DAFT_TREE=ad6f6fcb02ca6f6e7fbfac3bd762082ee93a2f62 TAO_CORE_COMMIT=dc721c75389c4917bb6f959b7f69620d7b90ce04 TAO_CORE_TREE=8b3d53453b43559ffcf69b92392cabfff6cf84d4 COSMOS_RL_COMMIT=f369c5c73af69212b9f1bb5e026888328f49a8fd COSMOS_RL_TREE=ad622ec8fbfde8e8fb6697f7e9c8943294a1a2c8 CUDA_COMPAT_LIBRARY=/usr/local/cuda/compat/lib.real CUDA_COMPAT_DRIVER_VERSION=595.45.04 DEEPEP_COMMIT=567632d NVSHMEM_PY_VERSION=3.4.5 USE_LOCAL_COSMOS_RL_GITHUB=true 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= 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
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    RUN_CLI=0
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    FLASK_APP=nvidia_tao_core.microservices.app
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4EXPOSE
    [8000/tcp]
    08/25/2026 6:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com
    08/25/2026 6:31 PM UTC
    ...

    NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.