Skip to content

Commit eb0588f

Browse files
authored
Update dockefile to keep up with latest version (#779)
* Update dockefile to keep up with latest version * Update CMake version * Update submodule version
1 parent 8231491 commit eb0588f

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

dockerfile/Dockerfile.triton.trt_llm_backend

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,18 @@
1-
ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:25.04-py3-min
2-
ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.04-py3
1+
ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:25.05-py3-min
2+
ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.05-py3
33
ARG NVRTC_VER=12.9.41-1
44
ARG TRT_VER=10.10.0.31
55
ARG RELEASE_URL_TRT_x86=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.10.0/tars/TensorRT-${TRT_VER}.Linux.x86_64-gnu.cuda-12.9.tar.gz
66
ARG RELEASE_URL_TRT_ARM=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.10.0/tars/TensorRT-${TRT_VER}.Linux.aarch64-gnu.cuda-12.9.tar.gz
77

88
# Versions of packages to copy from pytorch image
9-
ARG TORCH_VER=2.7.0a0+79aa17489c.nv25.4
9+
ARG TORCH_VER=2.8.0a0+5228986c39.nv25.5
1010
ARG TORCHVISION_VER=0.22.0a0
11-
ARG SETUPTOOLS_VER=78.1.0
12-
ARG PYTORCH_TRITON_VER=3.2.0+git4b3bb1f8b.nvinternal
11+
ARG SETUPTOOLS_VER=78.1.1
12+
ARG PYTORCH_TRITON_VER=3.3.0+git96316ce52.nvinternal
1313
ARG JINJA2_VER=3.1.6
1414
ARG NETWORKX_VER=3.4.2
15-
ARG SYMPY_VER=1.13.3
15+
ARG SYMPY_VER=1.14.0
1616
ARG PACKAGING_VER=23.2
1717
ARG FLASH_ATTN_VER=2.7.3
1818

@@ -137,7 +137,8 @@ ARG TENSORRTLLM_REPO_TAG
137137
RUN git-lfs install \
138138
&& git clone --single-branch --recurse-submodules --depth=1 -b ${TENSORRTLLM_REPO_TAG} ${TENSORRTLLM_REPO} tensorrt_llm
139139

140-
RUN pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 cmake==3.30.2 ninja
140+
RUN pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 cmake==4.0.3 ninja
141+
ENV CMAKE_POLICY_VERSION_MINIMUM=3.5
141142

142143
RUN cd tensorrt_llm && \
143144
if [ -n "${CCACHE_REMOTE_STORAGE}" ] ; then \

0 commit comments

Comments
 (0)