diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 0e003ca3d..212b965b5 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -3,7 +3,7 @@ numpy < 1.24; platform_machine == 'x86_64' tensorrt == 8.6.1; platform_machine == 'x86_64' tensorrt_bindings == 8.6.1; platform_machine == 'x86_64' -cuda-python == 11.8.*; platform_machine == 'x86_64' +cuda-python == 12.9.*; platform_machine == 'x86_64' cython == 3.0.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu11 == 11.8.*; platform_machine == 'x86_64' diff --git a/docker/tensorrt/requirements-arm64.txt b/docker/tensorrt/requirements-arm64.txt index c9b618180..fa3409bf4 100644 --- a/docker/tensorrt/requirements-arm64.txt +++ b/docker/tensorrt/requirements-arm64.txt @@ -1 +1 @@ -cuda-python == 12.6.*; platform_machine == 'aarch64' +cuda-python == 12.9.*; platform_machine == 'aarch64'