NVIDIA
NVIDIA
TAO Toolkit
Container
NVIDIA
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
/bin/bash
05/16/2026 12:15 AM UTC
sha256:e621ad8352a938dc043b552579b5868971c8547abf2e8464e7d429e1e6245ca8RUN
INSTALL_DEV=true PIP_CONSTRAINT="" python -m pip install --upgrade "aiohttp>=3.13.3" "gitpython>=3.1.48" "urllib3>=2.6.3" "setuptools>=78.1.1" "wheel>=0.46.2" "onnx>=1.21.0" "protobuf>=5.29.6" "sentencepiece>=0.2.1" "jupyter-server>=2.18.0" "jupyterlab>=4.5.7" "nbconvert>=7.17.0" "notebook>=7.5.6" "mistune>=3.2.1" "tornado>=6.5.5"
05/16/2026 12:15 AM UTC
sha256:4dbecb5eb86c6fe0702009f7c40ab97d741a1963abcee12ddeadd0378af13b60RUN
INSTALL_DEV=true make build-wheel &&
  WHEEL_FILE=$(ls -t dist/*.whl | head -1) &&
  if [ "$INSTALL_DEV" = "true" ]; then PIP_CONSTRAINT="" pip install "${WHEEL_FILE}[dev]"; else PIP_CONSTRAINT="" pip install "${WHEEL_FILE}"; fi &&
  make clean
05/16/2026 12:15 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
INSTALL_DEV=false
05/16/2026 12:15 AM UTC
sha256:0855cbebb7e8327bff64f4da7cd0b4a5845fcdb06d0921a199e488d889c6b60bCOPY
. /workspace/
05/16/2026 12:14 AM UTC
sha256:c1e18d89aac51a2adaa16c89bc27f2f9ae267b2bf9ae6dc0ab1b3c7c1090715aRUN
apt-get update &&
  apt --only-upgrade install git -y &&
  apt --only-upgrade install git-man -y &&
  apt --only-upgrade install gpg-agent -y &&
  apt --only-upgrade install gpgconf -y &&
  apt --only-upgrade install gpgsm -y &&
  apt --only-upgrade install gpgv -y &&
  apt --only-upgrade install keyboxd -y &&
  apt --only-upgrade install libtiff6 -y &&
  rm -rf /var/lib/apt/lists/*
05/16/2026 12:14 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace
05/16/2026 12:14 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
/bin/bash
07/23/2025 11:20 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace
07/23/2025 11:20 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
TARGETPLATFORM=linux/amd64 NATTEN_CUDA_ARCH=8.0;8.6;8.9;9.0;10.0;10.3;12.0 mkdir -p /workspace
07/23/2025 11:20 PM UTC
...