diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index a7853aeec..e8d39b308 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -11,7 +11,7 @@ nvidia-cuda-nvrtc-cu12==12.5.82; platform_machine == 'x86_64' nvidia_cuda_runtime_cu12==12.5.82; platform_machine == 'x86_64' nvidia_cusolver_cu12==11.6.3.*; platform_machine == 'x86_64' nvidia_cusparse_cu12==12.5.1.*; platform_machine == 'x86_64' -nvidia_nccl_cu12==2.23.4; platform_machine == 'x86_64' +nvidia_nccl_cu12==2.28.3; platform_machine == 'x86_64' nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64' tensorflow==2.19.*; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64'