NVIDIA
TAO Toolkit
Container
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
    ...