|
71 | 71 | # |
72 | 72 |
|
73 | 73 | DEFAULT_TRITON_VERSION_MAP = { |
74 | | - "release_version": "2.63.0dev", |
| 74 | + "release_version": "2.64.0dev", |
75 | 75 | "triton_container_version": "25.11dev", |
76 | | - "upstream_container_version": "25.10", |
| 76 | + "upstream_container_version": "25.11", |
77 | 77 | "ort_version": "1.23.2", |
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.2", |
| 81 | + "vllm_version": "0.11.0", |
82 | 82 | "rhel_py_version": "3.12.3", |
83 | 83 | } |
84 | 84 |
|
@@ -1500,6 +1500,15 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach |
1500 | 1500 |
|
1501 | 1501 | ARG PYVER=3.12 |
1502 | 1502 | ENV LD_LIBRARY_PATH /usr/local/lib:/usr/local/lib/python${{PYVER}}/dist-packages/torch/lib:${{LD_LIBRARY_PATH}} |
| 1503 | +""" |
| 1504 | + if "tensorrtllm" in backends or "vllm" in backends: |
| 1505 | + df += """ |
| 1506 | +ENV TRITON_CUDACRT_PATH=/usr/local/cuda/include \\ |
| 1507 | + TRITON_CUDART_PATH=/usr/local/cuda/include \\ |
| 1508 | + TRITON_CUOBJDUMP_PATH=/usr/local/cuda/bin/cuobjdump \\ |
| 1509 | + TRITON_CUPTI_PATH=/usr/local/cuda/include \\ |
| 1510 | + TRITON_NVDISASM_PATH=/usr/local/cuda/bin/nvdisasm \\ |
| 1511 | + TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas |
1503 | 1512 | """ |
1504 | 1513 |
|
1505 | 1514 | if "dali" in backends: |
|
0 commit comments