NVIDIA
NVIDIA
TAO Toolkit
Container
NVIDIA
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

LayerLabelCreated
cd4386734e84b0a533f651a7134aab26e69ce7dc965adcc355a86fd62cc69c98CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
05/16/2026 12:15 AM UTC
f66c766ac70bcd443ab134065ba775d01f427df73a4ff40d6d786d072e1f5881RUN
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
304749a10c5c5605f49d45165fee87d7525a280fd18f19c74ce9630f631f68cdRUN
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
1c768094c9086d92d134740434c5529717d33672dd5c0ecf3e1bb64181a48378ARG
INSTALL_DEV=false
05/16/2026 12:15 AM UTC
24f6a495d7484839a9725eaaa8476f2f42cea4edb60f3c082eb4708945044571COPY
. /workspace/
05/16/2026 12:14 AM UTC
17e023d29425f1b3130c02761c637330984811c2ac5aee08d8ca1e88b6e88dd7RUN
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
2eb4a68c1b4f5fa3026f1b69f8bfe05eb3c741def66b4a7bd09129738b77832fWORKDIR
/workspace
05/16/2026 12:14 AM UTC
ec0fc4fafc248aec2966cbd567d2c1338e974abef238ad6a838e42add752d8e4CMD
/bin/bash
07/23/2025 11:20 PM UTC
79544c8347ea9ffd59dae9c2c1d0fc70ab1fce781e3eb1db3b2b2e0791a7c74eWORKDIR
/workspace
07/23/2025 11:20 PM UTC
debce96243ddf56135b2212642e1d27d0b44d5e3632206f27647bb1cd1d9ef3dRUN
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.