File tree Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Original file line number Diff line number Diff line change 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
33ARG NVRTC_VER=12.9.41-1
44ARG TRT_VER=10.10.0.31
55ARG 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
66ARG 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
1010ARG 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
1313ARG JINJA2_VER=3.1.6
1414ARG NETWORKX_VER=3.4.2
15- ARG SYMPY_VER=1.13.3
15+ ARG SYMPY_VER=1.14.0
1616ARG PACKAGING_VER=23.2
1717ARG FLASH_ATTN_VER=2.7.3
1818
@@ -137,7 +137,8 @@ ARG TENSORRTLLM_REPO_TAG
137137RUN 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
142143RUN cd tensorrt_llm && \
143144 if [ -n "${CCACHE_REMOTE_STORAGE}" ] ; then \
You can’t perform that action at this time.
0 commit comments