NVIDIA
NVIDIA
TAO Toolkit
Container
NVIDIA
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

LayerLabelCreated
83a633b4a71da0b1f196603586a46c186f7a3d0ff2b046a9c283e1bf2e5ccdc5CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/
08/21/2024 3:40 AM UTC
983a4f146330c45cd242406a74c0d57dcbfb35cd72767f6b7bbfdaf80902ebcdRUN
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
3c09eaf6e82e7dc26e633fda09be81041e9cd080673d47b095d106bcc404d99eARG
home=/home/developer
08/21/2024 3:40 AM UTC
b5c1a1890b4673f284fc22a8a14f4b36122b27995a028ff61eb624b33dccaac4ARG
groups=developer:1000
08/21/2024 3:40 AM UTC
2ac2a0c2e28ed4e5bc9cb3c31d76fff65bea9334151ec3def38b5889ca88ff06ARG
user_name=developer
08/21/2024 3:40 AM UTC
ebfb2705507fbdfaf19571967de9da57329915d483a2df3db81a987b347e7e75ARG
user_id=1000
08/21/2024 3:40 AM UTC
ee6f494ccf3f6a30e6d34d76b9cdbc4555a1e6583100884403d8069e078332a7WORKDIR
/workspace/
08/21/2024 3:40 AM UTC
183b645c7d8c259b17d0303d24c868b52987dc5399a626400329d6691d4467d7RUN
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
12fe8492fbb07cb6943476b06c9784b1273e457578b23ec64d2a2c205ffc66a8COPY
tao-core tao-core
08/21/2024 3:39 AM UTC
65b2e4236217ba511645f14a46fdaa8033c88dd3c742b595ba187452cbae3664RUN
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
...