File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 11ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:25.06-py3-min
22ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.06-py3
33ARG NVRTC_VER=12.9.41-1
4- ARG TRT_VER=10.11 .0.33
4+ ARG TRT_VER=10.13 .0.35
55ARG NCCL_VER=2.27.5-1+cuda12.9
6- ARG RELEASE_URL_TRT_x86=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11 .0/tars/TensorRT-${TRT_VER}.Linux.x86_64-gnu.cuda-12.9.tar.gz
7- ARG RELEASE_URL_TRT_ARM=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11 .0/tars/TensorRT-${TRT_VER}.Linux.aarch64-gnu.cuda-12.9.tar.gz
6+ ARG RELEASE_URL_TRT_x86=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13 .0/tars/TensorRT-${TRT_VER}.Linux.x86_64-gnu.cuda-12.9.tar.gz
7+ ARG RELEASE_URL_TRT_ARM=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13 .0/tars/TensorRT-${TRT_VER}.Linux.aarch64-gnu.cuda-12.9.tar.gz
88
99# Versions of packages to copy from pytorch image
1010ARG FLASH_ATTN_VER=2.7.4.post1
You can’t perform that action at this time.
0 commit comments