CUDNN lib path fix in tensorrt docker

Signed-off-by: Riyad Islam <[email protected]>

Author: i-riyad

examples/onnx_ptq/docker/Dockerfile

@@ -14,8 +14,9 @@ WORKDIR /workspace
 
 RUN pip install tensorrt==10.13.2.6 && \
     export TRT_PATH=$(python -c "import tensorrt; import os; print(os.path.dirname(tensorrt.__file__))") && \
-    export LD_LIBRARY_PATH="$TRT_PATH/lib:/usr/include:${LD_LIBRARY_PATH}" && \
-    export PATH="$TRT_PATH/bin:${PATH}"
+    export CUDNN_LIB_DIR=$(python -c "import site; print(site.getsitepackages()[0])")/nvidia/cudnn/lib && \
+    export LD_LIBRARY_PATH="${CUDNN_LIB_DIR}:${TRT_PATH}/lib:/usr/include:${LD_LIBRARY_PATH}" && \
+    export PATH="${TRT_PATH}/bin:${PATH}"
 
 # Copy application code and install requirements
 COPY modelopt modelopt/modelopt