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
...

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.