diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 597680c00..8f3250023 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -11,7 +11,7 @@ nvidia-cuda-nvrtc-cu12==12.8.93; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12==12.8.90; platform_machine == 'x86_64' nvidia-cusolver-cu12==11.7.3.90; platform_machine == 'x86_64' nvidia-cusparse-cu12==12.5.8.93; platform_machine == 'x86_64' -nvidia-nccl-cu12==2.26.2.post1; platform_machine == 'x86_64' +nvidia-nccl-cu12==2.30.3; platform_machine == 'x86_64' nvidia-nvjitlink-cu12==12.8.93; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' onnxruntime-gpu==1.24.*; platform_machine == 'x86_64'