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