|
71 | 71 | # |
72 | 72 |
|
73 | 73 | DEFAULT_TRITON_VERSION_MAP = { |
74 | | - "release_version": "2.63.0dev", |
75 | | - "triton_container_version": "25.11dev", |
76 | | - "upstream_container_version": "25.09", |
| 74 | + "release_version": "2.62.0", |
| 75 | + "triton_container_version": "25.10", |
| 76 | + "upstream_container_version": "25.10", |
77 | 77 | "ort_version": "1.23.1", |
78 | 78 | "ort_openvino_version": "2025.3.0", |
79 | 79 | "standalone_openvino_version": "2025.3.0", |
80 | 80 | "dcgm_version": "4.4.0-1", |
81 | | - "vllm_version": "0.10.1.1", |
| 81 | + "vllm_version": "0.10.2", |
82 | 82 | "rhel_py_version": "3.12.3", |
83 | 83 | } |
84 | 84 |
|
@@ -660,6 +660,10 @@ def pytorch_cmake_args(images): |
660 | 660 | cargs.append( |
661 | 661 | cmake_backend_enable("pytorch", "TRITON_ENABLE_NVTX", FLAGS.enable_nvtx) |
662 | 662 | ) |
| 663 | + if target_platform() == "igpu": |
| 664 | + cargs.append( |
| 665 | + cmake_backend_enable("pytorch", "TRITON_PYTORCH_NVSHMEM", False) |
| 666 | + ) |
663 | 667 | return cargs |
664 | 668 |
|
665 | 669 |
|
@@ -1278,11 +1282,6 @@ def create_dockerfile_linux( |
1278 | 1282 | pip3 install --no-cache-dir grpcio-tools==1.64.0 && \\ |
1279 | 1283 | pip3 uninstall -y setuptools |
1280 | 1284 | ENV LD_LIBRARY_PATH=/usr/local/tensorrt/lib/:/opt/tritonserver/backends/tensorrtllm:$LD_LIBRARY_PATH |
1281 | | -
|
1282 | | -# There are some ucc issues when spawning mpi processes with ompi v4.1.7a1. |
1283 | | -# Downgrade to ompi v4.1.5rc2 to avoid the issue. |
1284 | | -RUN rm -fr /opt/hpcx/ompi |
1285 | | -COPY --from=nvcr.io/nvidia/tritonserver:24.02-py3-min /opt/hpcx/ompi /opt/hpcx/ompi |
1286 | 1285 | """ |
1287 | 1286 | with open(os.path.join(ddir, dockerfile_name), "w") as dfile: |
1288 | 1287 | dfile.write(df) |
@@ -1461,14 +1460,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach |
1461 | 1460 | virtualenv \\ |
1462 | 1461 | && rm -rf /var/lib/apt/lists/* |
1463 | 1462 | """ |
1464 | | - if "tensorrtllm" in backends: |
1465 | | - df += """ |
1466 | | -# Updating the openssh-client to fix for the CVE-2024-6387. This can be removed when trtllm uses a later CUDA container(12.5 or later) |
1467 | | -RUN apt-get update \\ |
1468 | | - && apt-get install -y --no-install-recommends \\ |
1469 | | - openssh-client \\ |
1470 | | - && rm -rf /var/lib/apt/lists/* |
1471 | | - """ |
1472 | 1463 |
|
1473 | 1464 | if "vllm" in backends: |
1474 | 1465 | df += f""" |
@@ -1514,6 +1505,25 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach |
1514 | 1505 | ENV PYTHONPATH=/opt/tritonserver/backends/dali/wheel/dali:$PYTHONPATH |
1515 | 1506 | """ |
1516 | 1507 |
|
| 1508 | + if ( |
| 1509 | + target_platform() not in ["igpu", "windows", "rhel"] |
| 1510 | + and "tensorrtllm" not in backends |
| 1511 | + ): |
| 1512 | + repo_arch = "sbsa" if target_machine == "aarch64" else "x86_64" |
| 1513 | + df += f""" |
| 1514 | +RUN curl -o /tmp/cuda-keyring.deb \\ |
| 1515 | + https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/{repo_arch}/cuda-keyring_1.1-1_all.deb \\ |
| 1516 | + && apt install /tmp/cuda-keyring.deb \\ |
| 1517 | + && rm /tmp/cuda-keyring.deb \\ |
| 1518 | + && apt update -qq \\ |
| 1519 | + && apt install --yes --no-install-recommends libnvshmem3-cuda-13 \\ |
| 1520 | + && rm -rf /var/lib/apt/lists/* \\ |
| 1521 | + && dpkg -L libnvshmem3-cuda-13 | grep libnvshmem_host.so | sed -e 's/libnvshmem_host.*//g' | sort -u > /etc/ld.so.conf.d/libnvshmem3-cuda-13.conf \\ |
| 1522 | + && ldconfig |
| 1523 | +""".format( |
| 1524 | + repo_arch=repo_arch |
| 1525 | + ) |
| 1526 | + |
1517 | 1527 | df += """ |
1518 | 1528 | WORKDIR /opt/tritonserver |
1519 | 1529 | RUN rm -fr /opt/tritonserver/* |
|
0 commit comments