diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 63c68b583..f5483adcb 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -3,7 +3,7 @@ cython==3.0.*; platform_machine == 'x86_64' nvidia_cuda_cupti_cu12==12.5.82; platform_machine == 'x86_64' nvidia-cublas-cu12==12.5.3.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12==9.3.0.*; platform_machine == 'x86_64' +nvidia-cudnn-cu12==9.4.0.*; platform_machine == 'x86_64' nvidia-cufft-cu12==11.2.3.*; platform_machine == 'x86_64' nvidia-curand-cu12==10.3.6.*; platform_machine == 'x86_64' nvidia_cuda_nvcc_cu12==12.5.82; platform_machine == 'x86_64'