diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index a7853aeec..7346ce742 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -16,4 +16,4 @@ nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64' tensorflow==2.19.*; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' onnxruntime-gpu==1.22.*; platform_machine == 'x86_64' -protobuf==3.20.3; platform_machine == 'x86_64' +protobuf==6.32.1; platform_machine == 'x86_64' diff --git a/docker/tensorrt/requirements-models-arm64.txt b/docker/tensorrt/requirements-models-arm64.txt index fe89b4754..e698e5aeb 100644 --- a/docker/tensorrt/requirements-models-arm64.txt +++ b/docker/tensorrt/requirements-models-arm64.txt @@ -1,2 +1,2 @@ onnx == 1.14.0; platform_machine == 'aarch64' -protobuf == 3.20.3; platform_machine == 'aarch64' +protobuf == 6.32.1; platform_machine == 'aarch64'