diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 597680c00..5057ae137 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -2,7 +2,7 @@ --extra-index-url 'https://pypi.nvidia.com' cython==3.0.*; platform_machine == 'x86_64' nvidia-cuda-cupti-cu12==12.8.90; platform_machine == 'x86_64' -nvidia-cublas-cu12==12.8.4.1; platform_machine == 'x86_64' +nvidia-cublas-cu12==12.9.2.10; platform_machine == 'x86_64' nvidia-cudnn-cu12==9.8.0.87; platform_machine == 'x86_64' nvidia-cufft-cu12==11.3.3.83; platform_machine == 'x86_64' nvidia-curand-cu12==10.3.9.90; platform_machine == 'x86_64'