|
1 | 1 | ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:25.10-py3-min |
2 | 2 | ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.10-py3 |
3 | 3 | ARG NVRTC_VER=13.0.48-1 |
4 | | -ARG TRT_VER=10.13.2.6 |
| 4 | +ARG TRT_VER=10.13.3.9 |
5 | 5 | ARG NCCL_VER=2.27.7-1+cuda13.0 |
6 | | -ARG RELEASE_URL_TRT_x86=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-${TRT_VER}.Linux.x86_64-gnu.cuda-13.0.tar.gz |
7 | | -ARG RELEASE_URL_TRT_ARM=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-${TRT_VER}.Linux.aarch64-gnu.cuda-13.0.tar.gz |
| 6 | +ARG RELEASE_URL_TRT_x86=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.3/tars/TensorRT-${TRT_VER}.Linux.x86_64-gnu.cuda-13.0.tar.gz |
| 7 | +ARG RELEASE_URL_TRT_ARM=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.3/tars/TensorRT-${TRT_VER}.Linux.aarch64-gnu.cuda-13.0.tar.gz |
8 | 8 |
|
9 | 9 | # Versions of packages to copy from pytorch image |
10 | | -ARG TORCH_VER=2.8.0a0+34c6371d24.nv25.8 |
11 | | -ARG TORCHVISION_VER=0.23.0a0+428a54c9 |
12 | | -ARG SETUPTOOLS_VER=79.0.1 |
13 | | -ARG PYTORCH_TRITON_VER=3.3.1+gitc8757738 |
| 10 | +ARG FLASH_ATTN_VER=2.7.4.post1+25.10 |
14 | 11 | ARG NETWORKX_VER=3.5 |
| 12 | +ARG PACKAGING_VER=25.0 |
| 13 | +ARG PYTORCH_TRITON_VER=3.4.0+gitc817b9b6 |
| 14 | +ARG SETUPTOOLS_VER=79.0.1 |
15 | 15 | ARG SYMPY_VER=1.14.0 |
16 | | -ARG PACKAGING_VER=23.2 |
17 | | -ARG FLASH_ATTN_VER=2.7.4.post1 |
| 16 | +ARG TORCH_VER=2.9.0a0+145a3a7bda.nv25.10 |
| 17 | +ARG TORCHVISION_VER=0.24.0a0+094e7af5 |
18 | 18 |
|
19 | 19 | ARG TENSORRTLLM_REPO=https://github.com/NVIDIA/TensorRT-LLM.git |
20 | | -ARG TENSORRTLLM_REPO_TAG=v1.2.0rc0 |
21 | | -ARG TENSORRTLLM_VER=1.2.0rc0 |
| 20 | +ARG TENSORRTLLM_REPO_TAG=v1.2.0rc2 |
| 21 | +ARG TENSORRTLLM_VER=1.2.0rc2 |
22 | 22 |
|
23 | 23 | FROM ${PYTORCH_IMAGE} AS pytorch_image |
24 | 24 | FROM ${BASE_IMAGE} AS install_dependencies |
|
0 commit comments