From a86e2a709cb930053f6a44fa0606aba1d960d181 Mon Sep 17 00:00:00 2001
From: Misha Chornyi <mchornyi@nvidia.com>
Date: Wed, 9 Jul 2025 11:02:49 -0700
Subject: [PATCH] build: TPRD-1622 - Update version to CMake 4.0 or 3.31 (
 3.31.8 ) which has support for CUDA new layout

---
 dockerfile/Dockerfile.triton.trt_llm_backend | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dockerfile/Dockerfile.triton.trt_llm_backend b/dockerfile/Dockerfile.triton.trt_llm_backend
index 787f09e..a2351f4 100644
--- a/dockerfile/Dockerfile.triton.trt_llm_backend
+++ b/dockerfile/Dockerfile.triton.trt_llm_backend
@@ -137,7 +137,7 @@ ARG TENSORRTLLM_REPO_TAG
 RUN git-lfs install \
     && git clone --single-branch --recurse-submodules --depth=1 -b ${TENSORRTLLM_REPO_TAG} ${TENSORRTLLM_REPO} tensorrt_llm
 
-RUN pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 cmake==3.30.2 ninja
+RUN pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 cmake==4.0.3 ninja
 
 RUN cd tensorrt_llm && \
     if [ -n "${CCACHE_REMOTE_STORAGE}" ] ; then \