CUDNN lib path fix in tensorrt docker

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

Author: i-riyad

examples/onnx_ptq/docker/Dockerfile

@@ -14,7 +14,8 @@ 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 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