From eaf2d998a43b20434ac742f078c365e1b8a00379 Mon Sep 17 00:00:00 2001
From: Kok Seang Tan <kseangtan@gmail.com>
Date: Wed, 16 Jul 2025 13:19:31 +0900
Subject: [PATCH] Experiment with tensorrt optimization

---
 Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index a53ee994c..e19746d54 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -49,12 +49,13 @@ RUN mim install \
     mmpretrain[multimodal]==${MMPRETRAIN} \
     mmsegmentation==${MMSEGMENTATION}
 
-# Install rerun
+# Install rerun & nvidia-modelopt (tensorrt model optimization)
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
     libgtk-3-dev \
     libxkbcommon-x11-0
 RUN python3 -m pip --no-cache-dir install \
-    rerun-sdk==0.17.0
+    rerun-sdk==0.17.0 \
+		nvidia-modelopt==0.33.0
 
 # Install t4-devkit
 RUN python3 -m pip install git+https://github.com/tier4/t4-devkit@v0.0.7