diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index df276a613..bfe2df363 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -11,4 +11,4 @@ nvidia-cudnn-cu11 == 8.6.0.*; platform_machine == 'x86_64' nvidia-cufft-cu11==10.*; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' onnxruntime-gpu==1.18.*; platform_machine == 'x86_64' -protobuf==3.20.3; platform_machine == 'x86_64' +protobuf==5.29.2; platform_machine == 'x86_64' diff --git a/docker/tensorrt/requirements-models-arm64.txt b/docker/tensorrt/requirements-models-arm64.txt index 3490a7897..9543c7aaf 100644 --- a/docker/tensorrt/requirements-models-arm64.txt +++ b/docker/tensorrt/requirements-models-arm64.txt @@ -1,3 +1,3 @@ onnx == 1.14.0; platform_machine == 'aarch64' -protobuf == 3.20.3; platform_machine == 'aarch64' +protobuf == 5.29.2; platform_machine == 'aarch64' numpy == 1.23.*; platform_machine == 'aarch64' # required by python-tensorrt 8.2.1 (Jetpack 4.6)