NVIDIA
NVIDIA
TAO Toolkit
Container
NVIDIA
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

LayerLabelCreated
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace/
08/21/2024 3:40 AM UTC
sha256:84cb9b8b8ad9adbcd60e3b57faf91b2fc7a9eec67b6952442e2fdd8c236aeedeRUN
TRT_VERSION_MAJOR=8 TRT_VERSION_MINOR=6 TRT_VERSION_PATCH=3 TRT_VERSION_BUILD=1 TRT_VERSION_MAJOR_MINOR=8.6 TRT_VERSION_MAJOR_MINOR_PATCH=8.6.3 TRT_VERSION_FULL=8.6.3.1 CUDA_VERSION_MAJOR=12 CUDA_VERSION_MINOR=2 CUDA_VERSION_PATCH=2 CUDA_VERSION_BUILD=009 CUDA_VERSION_MAJOR_MINOR=12.2 CUDA_VERSION_FULL=12.2.2.009 CUDNN_VERSION=9.0.0.306 user_id=1000 user_name=developer groups=developer:1000 home=/home/developer echo "ALL   ALL = (ALL) NOPASSWD: ALL" > /etc/sudoers &&
  mkdir -p "$(dirname $home)"
08/21/2024 3:40 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
home=/home/developer
08/21/2024 3:40 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
groups=developer:1000
08/21/2024 3:40 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
user_name=developer
08/21/2024 3:40 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
user_id=1000
08/21/2024 3:40 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace/
08/21/2024 3:40 AM UTC
sha256:bdc2e2ec3eb173f59069501649645263edbd97310d9ea673d7d5c8c08f7d1454RUN
TRT_VERSION_MAJOR=8 TRT_VERSION_MINOR=6 TRT_VERSION_PATCH=3 TRT_VERSION_BUILD=1 TRT_VERSION_MAJOR_MINOR=8.6 TRT_VERSION_MAJOR_MINOR_PATCH=8.6.3 TRT_VERSION_FULL=8.6.3.1 CUDA_VERSION_MAJOR=12 CUDA_VERSION_MINOR=2 CUDA_VERSION_PATCH=2 CUDA_VERSION_BUILD=009 CUDA_VERSION_MAJOR_MINOR=12.2 CUDA_VERSION_FULL=12.2.2.009 CUDNN_VERSION=9.0.0.306 cd tao-core &&
  bash release/python/build_wheel.sh &&
  find dist/ -name "nvidia_tao_core*.whl" -type f | xargs -n 1 pip install &&
  rm -rf *
08/21/2024 3:40 AM UTC
sha256:74aa553bf3723f7e1c98283efb3cacc6f8ac2b194fc28970bdb77ed80c6a567dCOPY
tao-core tao-core
08/21/2024 3:39 AM UTC
sha256:c9d44668f38d506fbc3a65cecb4531ca21cbab2de45a6f6552db6a87d8280081RUN
TRT_VERSION_MAJOR=8 TRT_VERSION_MINOR=6 TRT_VERSION_PATCH=3 TRT_VERSION_BUILD=1 TRT_VERSION_MAJOR_MINOR=8.6 TRT_VERSION_MAJOR_MINOR_PATCH=8.6.3 TRT_VERSION_FULL=8.6.3.1 CUDA_VERSION_MAJOR=12 CUDA_VERSION_MINOR=2 CUDA_VERSION_PATCH=2 CUDA_VERSION_BUILD=009 CUDA_VERSION_MAJOR_MINOR=12.2 CUDA_VERSION_FULL=12.2.2.009 CUDNN_VERSION=9.0.0.306 mkdir trt_oss_src &&
  cd trt_oss_src &&
  echo "$PWD Building TRT OSS..." &&
  git clone -b $TRT_TAG https://github.com/nvidia/TensorRT TensorRT &&
  cd TensorRT &&
  git submodule update --init --recursive &&
  mkdir -p build &&
  cd build &&
  cmake .. -DGPU_ARCHS="53;60;61;70;75;80;86;90" -DCMAKE_CUDA_ARCHITECTURES="53;60;61;70;75;80;86;90" -DTRT_LIB_DIR=/usr/lib/x86_64-linux-gnu -DTRT_BIN_DIR=`pwd`/out -DCUDA_VERSION=$CUDA_VERSION_MAJOR_MINOR -DCUDNN_VERSION=$CUDNN_VERSION &&
  make -j32 &&
  cp libnvinfer_plugin.so.8.6.* /usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.$TRT_VERSION_MAJOR_MINOR_PATCH &&
  cp libnvinfer_plugin_static.a /usr/lib/x86_64-linux-gnu/libnvinfer_plugin_static.a &&
  cp libnvonnxparser.so.8.6.* /usr/lib/x86_64-linux-gnu/libnvonnxparser.so.$TRT_VERSION_MAJOR_MINOR_PATCH &&
  cp libnvcaffeparser.so.8.6.* /usr/lib/x86_64-linux-gnu/libnvcaffeparser.so.$TRT_VERSION_MAJOR_MINOR_PATCH &&
  cp trtexec /usr/local/bin/ &&
  rm -rf trt_oss_src
08/21/2024 3:39 AM UTC
...